python - 删除特定列号后所有包含全零的行

Question

在 pandas dataframe 中，我如何删除在特定列之后有零的所有行。例如

from pandas import DataFrame
df = DataFrame({'a' : [0,1,1,0,0,0,0], 'b' : [0,1,-1, 1,0,0,0], 'c': [1,4,5,6,7,0,0]}).T

df:

    0   1   2   3   4   5   6
a   0   1   1   0   0   0   0
b   0   1   -1  1   0   0   0
c   1   4   5   6   7   0   0

如何在第 3 列之后将包含所有值的行删除为零？此示例中的第一行和第二行(索引 a 和 b)将被删除。

Answer 1

您可以为列添加下标，将 0 替换为 NaN，删除任何没有至少 1 个非 NaN 值的行，然后在索引上使用 loc:

In [63]:
df.loc[df[df.columns[4:]].replace(0, NaN).dropna(thresh=1).index]
Out[63]:
   0  1  2  3  4  5  6
c  1  4  5  6  7  0  0

所以分解一下:

In [64]:
df[df.columns[4:]]

Out[64]:
   4  5  6
a  0  0  0
b  0  0  0
c  7  0  0

In [66]:   
df[df.columns[4:]].replace(0, NaN)

Out[66]:
    4   5   6
a NaN NaN NaN
b NaN NaN NaN
c   7 NaN NaN

In [67]:    
df[df.columns[4:]].replace(0, NaN).dropna(thresh=1)

Out[67]:
   4   5   6
c  7 NaN NaN

In [68]:    
df[df.columns[4:]].replace(0, NaN).dropna(thresh=1).index

Out[68]:
Index(['c'], dtype='object')

更新其实更简洁的方式:

In [77]:

df[any(df[df.columns[4:]] != 0, axis=1)]
Out[77]:
   0  1  2  3  4  5  6
c  1  4  5  6  7  0  0