python - 我想找到我的数据中某些列的最大值不是1

Question

我有 254 行 x 474 列的 DataFrame 数据。

我想找到列的最大值不是1的列。大多数列的最大值是1，但有些列的最大值不是1。

我使用python 3.x环境。

import pandas as pd

data = pd.read_csv("data.csv") # this is my data.

Answer 1

使用DataFrame.loc使用 : 获取所有行和 DataFrame.max对于过滤列:

df = pd.DataFrame({'a':[0,2,1],
                   'b':[0,0,1],
                   'c':[3,2,1]})

df1 = df.loc[:, df.max() != 1]
print (df1)
   a  c
0  0  3
1  2  2
2  1  1

如果仅需要列名称:

c = df.columns[df.max() != 1]
print (c)
Index(['a', 'c'], dtype='object')