python - 如何根据给定的行值保留列

Question

df 数据帧中的数据如下所示:

        A   B   C   D
0.js    2   1   1  -1
1.js    3  -5   1  -4
total   5  -4   2  -5

我会得到新的数据框df1:

        A     C
0.js    2     1
1.js    3     1
total   5     2

所以基本上它应该是这样的: df1 = df[df["总计"] > 0] 但它应该在行而不是列上进行过滤，我无法弄清楚..

Answer 1

您想使用 .loc[:, column_mask] 即

In [11]: df.loc[:, df.sum() > 0]
Out[11]:
       A  C
total  5  2

# or

In [12]: df.loc[:, df.iloc[0] > 0]
Out[12]:
       A  C
total  5  2