熊貓：獲取列的最大值== 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']