需要使用 dataframe 列中存在的列表值创建列表

Question

这是我的 dataframe：

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

L=[Prod2,Prod4,Prod5]

列表中存在的产品的 ID

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

Answer 1

如果顺序是重要的列表理解，则使用：

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']