How to keep columns based on a given row values

Question

Here how the datalooks like in df dataframe:

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

And I would get new dataframe df1:

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

So basically it should look like this: df1 = df[df["total"] > 0] but it should filter on row instead of column and I can't figure it out..

Answer 1

You want to use .loc[:, column_mask] i.e.

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