如果范围列的值为0,则删除Pandas数据框的行

时间:2020-03-23 08:30:08

标签: python pandas python-2.7 dataframe

数据框:

df = pd.DataFrame({'a':['NA','W','Q','M'], 'b':[0,0,4,2], 'c':[0,12,0,2], 'd':[22, 3, 34, 12], 'e':[0,0,3,6], 'f':[0,2,0,0], 'h':[0,1,1,0] })
df
    a   b   c   d   e   f   h
0   NA  0   0   22  0   0   0
1   W   0   12  3   0   2   1
2   Q   4   0   34  3   0   1
3   M   2   2   12  6   0   0

如果列b和所有列e的值都包含0,我想删除整行

基本上我想得到这样的东西

a   b   c   d   e   f   h
1   W   0   12  3   0   2   1
2   Q   4   0   34  3   0   1
3   M   2   2   12  6   0   0

1 个答案:

答案 0 :(得分:1)

如果要测试从e到结尾列,以及由DataFrame.assign添加的b列,请使用DataFrame.loc进行选择,请测试不等于DataFrame.ne,然后如果aby值与DataFrame.any相匹配(表示没有所有0),并且最后由boolean indexing进行过滤:

df = df[df.loc[:, 'e':].assign(b = df['b']).ne(0).any(axis=1)]
print (df)
   a  b   c   d  e  f  h
1  W  0  12   3  0  2  1
2  Q  4   0  34  3  0  1
3  M  2   2  12  6  0  0