Создайте список имен индексов для кадра данных pandas, если условие выполнено - PullRequest
0 голосов
/ 06 августа 2020

У меня есть pandas фрейм данных, который выглядит так:

    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]

1 Ответ

0 голосов
/ 06 августа 2020

Вы можете использовать stack после транспонирования вашего набора данных:

df1 = df.T.stack()
df1[df 1>= 80].reset_index().level_1.values
['f' 'e' 'f' 'e' 'g' 'h']
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...