python – 选择列中只有零的行

2023年8月20日 252次阅读

我想选择数据框中的行,这些行在列列表的每列中都为零.例如这个df：

In: 
    df = pd.DataFrame([[1,2,3,6], [2,4,6,8], [0,0,3,4],[1,0,3,4],[0,0,0,0]],columns =['a','b','c','d'])
    df

Out:
    a   b   c   d
0   1   2   3   6
1   2   4   6   8
2   0   0   3   4
3   1   0   3   4
4   0   0   0   0

然后：

In:
    mylist = ['a','b']
    selection = df.loc[df['mylist']==0]
    selection

我想看看：

Out:      
    a   b   c   d
2   0   0   3   4
4   0   0   0   0

应该很简单,但我的日子很慢！

最佳答案 您需要确定一行中的所有列是否都为零.给定一个布尔掩码,使用DataFrame.all(axis = 1)来做到这一点.

df[df[mylist].eq(0).all(1)]

   a  b  c  d
2  0  0  3  4
4  0  0  0  0

请注意,如果要在每列中查找包含零的行,请删除子集步骤：

df[df.eq(0).all(1)]

   a  b  c  d
4  0  0  0  0
点赞