提取dataframe的前两列

您可以使用 Pandas 库中的 iloc 函数来提取 dataframe 的前两列，示例代码如下：

import pandas as pd

# 假设 df 是您的 dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 使用 iloc 函数提取前两列
df_new = df.iloc[:, :2]

# 打印提取后的 dataframe
print(df_new)

输出结果为：

   A  B
  1  4
1  2  5
2  3  6

相关问题

python dataframe 呈现两列

回答: 若要呈现DataFrame中的两列数据，可以使用matplotlib库中的scatter函数。该函数可以绘制散点图来展示两列数据之间的关系。下面是一个示例代码：

import pandas as pd
import matplotlib.pyplot as plt

path='IMDB-Movie-Data.csv'
movie_data=pd.read_csv(path)
pd.set_option('display.max_columns',None)

# 提取需要呈现的两列数据
data = movie_data[['Rating', 'Runtime']]

# 绘制散点图
plt.scatter(data['Rating'], data['Runtime'])
plt.xlabel('Rating')
plt.ylabel('Runtime')
plt.title('Rating vs Runtime')
plt.show()

以上代码会绘制一个以Rating为x轴，Runtime为y轴的散点图，用于展示这两列数据之间的关系。<em>1</em><em>2</em><em>3</em>
#### 引用[.reference_title]
- *1* *2* *3* [16、python中dataframe的合并行/列、分组与聚合、行索引](https://blog.csdn.net/weixin_45217685/article/details/131967358)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}} ] [.reference_item]
[ .reference_list ]

python如何提取dataframe中的两列作为一个列表

可以使用pandas库中的iloc方法来提取dataframe中的两列作为一个列表。具体代码如下：

import pandas as pd

# 创建一个示例dataframe
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 提取'A'和'B'列作为一个列表
lst = df.iloc[:, [, 1]].values.tolist()

print(lst)

输出结果为：

[[1, 4], [2, 5], [3, 6]]