首页 > 解决方案 > 如果满足条件,则创建熊猫数据框的索引名称列表

问题描述

我有一个看起来像的熊猫数据框:

    A   B   C
e  34  89  90
f  92  90  40
g  45  67  80
h  76  25  98

我想遍历列的每一行并检查值是否> 80,如果满足条件,我想提取索引名称并附加到列表中。最好的方法是什么?

结果列表应如下所示: ['f','e','f','e','g','h]

标签: python-3.xlistdataframefilterconditional-statements

解决方案


您可以stack在转置数据集后使用:

df1 = df.T.stack()
df1[df 1>= 80].reset_index().level_1.values
['f' 'e' 'f' 'e' 'g' 'h']

推荐阅读