DataFrame选取两列数据

可以使用 Pandas 的 loc 或 iloc 方法来选取 DataFrame 中的两列数据。

假设有一个 DataFrame 叫做 df，其中包含了三列数据，分别为 A、B、C，选取 A 和 B 两列数据的代码如下：

使用 loc 方法：

df.loc[:, ['A', 'B']]

使用 iloc 方法：

df.iloc[:, [0, 1]]

其中 loc 方法中的冒号表示选取所有行，中括号中的列表表示选取列名为 A 和 B 的两列数据。

而 iloc 方法中的冒号和列表分别代表选取所有行和选取第 1 和第 2 列数据（因为 Python 中的索引是从 0 开始的）。

选取完两列数据后，可以将结果赋值给新的 DataFrame 或者直接对选取结果进行其他的操作。

相关问题

选取dataframe中两列生成新的dataframe，索引与原dataframe相同

可以使用pandas的切片操作来选取dataframe中的两列，并生成一个新的dataframe。假设原dataframe的名称是df，我们要选取的两列分别是column1和column2，可以使用以下代码来实现：

new_df = df[['column1', 'column2']]

这样就会生成一个新的dataframe new_df，其中包含了原dataframe中的column1和column2两列数据，并且索引与原dataframe相同。

dataframe选取指定的两列数据

可以使用pandas的DataFrame的切片操作来选取指定的两列数据。以下是一个例子：

import pandas as pd

# 读取csv文件
data = pd.read_csv('data.csv')

# 选取指定的两列数据
selected_data = data[['column1', 'column2']]

其中，`data`是读取的csv文件的DataFrame对象，`column1`和`column2`是需要选取的两列数据的列名。选取后的结果存储在`selected_data`中，也是一个DataFrame对象。