熊猫：得到每个元素的索引

Question

我想这是Find element's index in pandas Series的副本。

这是我的数据框;

      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

您可以在Python中enumerate获取索引以及项目：

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是一种查找字符串中子字符串索引的方法。