保留数据帧pandas中的特定列

Question

我有一个使用pandas导入csv的数据帧。这个数据框有160个变量，我想只保留5,9,10,46,89。

我试试这个：

dataf2 = dataf[[5] + [9] + [10] + [46] + [89]]

但我接受了这个错误：

KeyError: '[ 5 9 10 46 89] not in index'

Answer 1

如果您要保留的列是5, 9, 10, 46, 89，那么您可以像这样索引这些列：

dataf2 = dataf[[5, 9, 10, 46, 89]]

Answer 2

如果您想要引用列不是通过名称而是通过它们在数据集中的位置，则需要使用df.iloc：

dataf.iloc[:, [5, 9, 10, 46, 89]]

在逗号之前指定行索引，在逗号后指定列索引。