我有一个数据框看起来像这样
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
试试这个:
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
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句