熊猫：获取列的最大值== N

Question

我知道这种方式的作品，但有没有更标准的方式做到这一点？

N=1 
cols = list(data.iloc[:,(data.max()==1).values].columns) 

Answer 1

这将做的工作：

df=pd.DataFrame({'a':[1,-1],'b':[3,4]}) 
df.columns[df.max()==1].tolist() 

这将打印['a']