根据python中的数据从数据框中提取列名

【问题标题】：extract column names from dataframe according to data in python根据python中的数据从数据框中提取列名
【发布时间】：2017-08-02 08:24:31
【问题描述】：

我有一个像这样的数据框

    a   b   c
a  0.3 0.2 0.9 
b  0.9   1 0.8
c  0.2 0.9 0.5

如果对应的值大于等于0.9，我想提取列名。

结果数据框如下所示：

a  c 
a  a b 
c  b

【问题讨论】：

【解决方案1】：

试试这个：

In [29]: df.ge(0.9).apply(lambda x: df.columns[x].tolist(), axis=1)
Out[29]:
a       [c]
b    [a, b]
c       [b]
dtype: object

【讨论】：