python – 选择列中只有零的行

发布时间：2020-12-20 11:05:07 所属栏目：Python 来源：网络整理

导读：我想选择数据框中的行,这些行在列列表的每列中都为零.例如这个df： In: df = pd.DataFrame([[1,2,3,6],[2,4,6,8],[0,4],[1,0]],columns =['a','b','c','d']) dfOut: a b c d0 1 2 3 61 2 4 6 82 0 0 3 43 1 0 3 44 0 0 0 0 然后： In: mylist = ['a','b'] se

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

In: 
    df = pd.DataFrame([[1,2,3,6],[2,4,6,8],[0,4],[1,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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!