Найти индекс из нескольких строк - PullRequest
0 голосов
/ 16 мая 2018

Предположим, у меня есть фрейм данных с именем df: A_column B_column C_column 0 Apple 100 15 1 Banana 80 20 2 Orange 110 10 3 Apple 150 16 4 Apple 90 13

[Q] Как перечислить индекс [0,3,4] для Apple в A_column?

1 Ответ

0 голосов
/ 16 мая 2018

Вы можете просто передать индексы строк в виде списка на df.iloc

>>> df
  A_column  B_column  C_column
0    Apple       100        15
1   Banana        80        20
2   Orange       110        10
3    Apple       150        16
4    Apple        90        13

>>> df.iloc[[0,3,4]]
  A_column  B_column  C_column
0    Apple       100        15
3    Apple       150        16
4    Apple        90        13

РЕДАКТИРОВАТЬ: кажется, я неправильно понял ваши вопросы

Итак, вы хотите, чтобы список, содержащий номер индексастроки, содержащие «Apple», можно использовать df.index[df['A_column']=='Apple'].tolist()

>>> df.index[df['A_column']=='Apple'].tolist()
[0, 3, 4]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...