Как выбрать определенное значение из столбца на основе другого столбца с несколькими идентификаторами

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

Насколько я понимаю, вы пытаетесь получить имя питомца из его идентификатора?Если так, это также должно работать

Answer 2

Я думаю, вам нужно DataFrame.loc с преобразованием значений в список:

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