根据 python 中的数据从数据框中提取列名
extract column names from dataframe according to data in python
我的数据框看起来像这样
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
我的数据框看起来像这样
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