python - pandas - 如何使用 bool 系列删除 DataFrame 列？

Question

我有一个程序可以执行此操作:

df1 = pd.DataFrame(np.random.randn(100,5),
                 columns=['a','b','c','d','e',])

从那里，我尝试删除均值> 0的列

如果我做

s = df1.mean() > 0

那么 s，就我而言，是一个 pandas 系列，等于:

a    False
b    False
c     True
d     True
e    False
dtype: bool

但是我如何将其应用于 df1 以便删除列 c 和 d？

我对整个事情的处理方式是错误的吗？

Answer 1

试试这个:

df1[df1.columns[~s]]

这将消除 s 为 True 的所有列，即 c 和 d。

实现这一目标的另一种方法是:

df1.loc[:, ~s]