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

Question

我的数据框看起来像这样

<a class="button" tabindex="0">CALCULATE</a>

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

结果数据框如下所示：

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

Answer 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