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 或者直接对选取结果进行其他的操作。