Необходимо создать список, используя значения списка, которые присутствуют в столбце данных - PullRequest
Новая
       5

Необходимо создать список, используя значения списка, которые присутствуют в столбце данных

0 голосов
/ 14 апреля 2020

Это мой фрейм данных: 

prod_sheet:
    Product ID
0   Prod1   00P000000000101
1   Prod2   00P000000000105
2   Prod3   00P000000000109
3   Prod4   00P000000000119
4   Prod5   00P000000000120

L=[Prod2,Prod4,Prod5]

Идентификатор продукта, который присутствует в списке 

needed_list=[00P000000000105,00P000000000119,00P000000000120]

1 Ответ

0 голосов
/ 14 апреля 2020

Используйте, если порядок важен для понимания списка: 

L=['Prod5','Prod4','Prod3']
s = prod_sheet.set_index('Product')['ID']

needed_list = [s[p] for p in L]
print (needed_list)
['00P000000000120', '00P000000000119', '00P000000000109']

Если порядок не важен, используйте: 

needed_list = prod_sheet.loc[prod_sheet['Product'].isin(L), 'ID'].tolist()
print (needed_list)
'00P000000000109', '00P000000000119', '00P000000000120']
...