如果满足条件,则创建熊猫数据框的索引名称列表

时间:2020-08-05 22:59:48

标签: python-3.x list dataframe filter conditional-statements

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

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

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

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

1 个答案:

答案 0 :(得分:0)

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

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