python - 如何制作计数高于特定值的数据框？

Question

我有一个看起来像这样的 df:

df

      a   b   c   d
 0    1   0   0   1
 1    1   1   0   1
 2    0   1   1   1
 3    1   0   0   1

我尝试获取列数大于 2 的 df。但找不到解决方案。它应该看起来像这样:

     a    d
0    1    1
1    1    1
2    0    1
3    1    1

Answer 1

如果只有 1 和 0 值，请使用 DataFrame.loc与 boolean indexing ，第一个 : 用于匹配所有行:

df = df.loc[:, df.sum() > 2]
print (df)
   a  d
0  1  1
1  1  1
2  0  1
3  1  1

详细信息:

print (df.sum())
a    3
b    2
c    1
d    4
dtype: int64

print (df.sum() > 2)
a     True
b    False
c    False
d     True
dtype: bool

如果可能的话，其他值只需要计数1:

df = df.loc[:, df.eq(1).sum() > 2]