python - 如何删除 DataFrame 中除某些列之外的所有列？

Question

假设我有一个如下所示的 DataFrame:

a  b  c  d  e  f  g  
1  2  3  4  5  6  7
4  3  7  1  6  9  4
8  9  0  2  4  2  1

除了 a 和 b 之外，我将如何删除每一列？

这将导致:

a  b
1  2
4  3
8  9

我想要一种使用简单的代码行删除这些的方法，即删除 a 和 b 之外的所有列，因为假设我有 1000 列数据。

谢谢。

Answer 1

In [48]: df.drop(df.columns.difference(['a','b']), 1, inplace=True)
Out[48]:
   a  b
0  1  2
1  4  3
2  8  9

或:

In [55]: df = df.loc[:, df.columns.intersection(['a','b'])]

In [56]: df
Out[56]:
   a  b
0  1  2
1  4  3
2  8  9

PS 请注意 the most idiomatic Pandas way to do that was already proposed by @Wen :

df = df[['a','b']]

或

df = df.loc[:, ['a','b']]