Pandas: получить индекс каждого элемента

Question

Я думаю, это дубликат Find element's index in pandas Series.

Это мой dataframe;

 WORD1 CAT1 
    elephant animal 
     lion animal 
     tiger animal 
     hoopoe bird 
    hornbill bird 
    sunflower flower 
     rose flower 
    giraffe animal 
     zebra animal 
    sparrow bird 
     duck animal 

Я хотел бы получить индекс каждого элемента из «CAT1»;

Позвольте мне сказать так;

for d in data['CAT1']: 
    print data[data['CAT1'] == d].index[0] 
... 
0 
0 
0 
3 
3 
5 
5 
0 
0 
3 
0 

Приведенное выше возвращает индекс, но колеблется, когда есть дубликаты. Как мне это исправить?

Answer 1

Вы можете enumerate в Python, чтобы получить индексы вместе с деталями:

for i, d in enumerate(data['CAT1']): 
    print(i) 

Если вы хотите, чтобы выбрать из WORD1 по CAT1, вы можете zip их, например:

birds = [w for w, c in zip(data['WORD1'], data['CAT1']) if c == "bird")] 

Примечание: str.index - это метод поиска индекса подстроки внутри строки.