python - 如何从一列中获取另一列的值与列表匹配的值

Question

我们有一个数据框:

      column1  column2
0        A       h
1        B       l
2        C       p

和一个列表:

li = ['p', 'l']

如何使用列表查找 column2 中的值并按顺序返回相应的 column1 值？

期望的结果:

np.array(['C', 'B'])

我尝试了这个，但它没有保留顺序:

df.loc[df['column2'].isin(li)]['column1'].values

Answer 1

将column2设置为索引并使用loc :

df.set_index('column2').loc[li, 'column1'].values

# array(['C', 'B'], dtype=object)