如何在数据帧中选择特定列？[副本]

Question

我有一个25列的数据帧和一个数组（[2,4,8,10,11,12,13,14,17,18,19,20,21,22,23]，dtype=int64）。我想从dataframe中选择一些特定的列，这些列的索引由数组的元素给出，并将它们存储在一个新的dataframe中（比如df1）。例如，新数据帧将有原始数据帧的第2、第4、第23列。

Answer 1

您可以使用iloc来完成此操作。例如：

import pandas as pd

df = pd.DataFrame(
    [np.random.rand(5),np.random.rand(5), np.random.rand(5), np.random.rand(5)]
)

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

哪些产出：

    1           3
0   0.883848    0.409460
1   0.537549    0.426643
2   0.825185    0.361043
3   0.039343    0.674435

您可能会看到较旧的答案建议.ix（例如我在这个问题中改编的答案），但是，这个答案已被弃用。