代码之家  ›  专栏  ›  技术社区  ›  pylearner

如何根据给定多个ID的另一列从列中选取特定值

  •  0
  • pylearner  · 技术社区  · 6 年前

    id

    :

    ID  A   B
    1   cat 22
    2   dog 33
    3   mamal   44
    4   rat 55
    5   rabbit  66
    6   puppy   77
    

    ID ,我需要该列的特定值。

    例子 :

    animal = []
    id = [1,2,3] 
    for i in id:
       if ID == i:
          animal.append(data.A[where ID == i])
    

    :

    dog
    
    2 回复  |  直到 6 年前
        1
  •  2
  •   jezrael    6 年前

    我想你需要 DataFrame.loc 将值转换为列表:

    animal = data.loc[data.ID == 2, 'A'].values.tolist()
    print (animal)
    ['dog']
    
        2
  •  0
  •   mate00 GMc    5 年前

    据我所知,你是想从宠物ID中检索一个宠物名字?如果是这样的话,这也应该起作用。

    pet = df.A.values[df.ID==2]
    

    这将导致宠物=狗