我想在我的数据中找到一些列,最大值不为1

时间:2019-07-07 13:51:11

标签: python pandas

我有254行x 474列DataFrame数据。

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

我使用python 3.x环境。

import pandas as pd

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

1 个答案:

答案 0 :(得分: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')