如何根据具有多个ID的另一列从一列中选择特定值

Question

我正在尝试从给定行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

Answer 1

我认为您需要DataFrame.loc并将值转换为列表：

animal = data.loc[data.ID == 2, 'A'].values.tolist()
print (animal)
['dog']

Answer 2

据我了解，您正在尝试从宠物ID中检索宠物名称？如果是这样，这也应该起作用。

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

这将导致pet = dog