如何用索引列表索引大熊猫的数据？

Question

我的数据就像美女一样。

df
   a  b  c
w  5  3  3
x  4  7  6
y  6  2  5
z  2  6  2

我有一份清单就像贝洛斯一样。

a
[['w', 'x', 'w'], 
['x', 'y', 'y']]

我怎样才能得到这样的结果

[[5,7,3],
[4,2,5]]

如果唯一的方法是使用for循环，哪种方法是最快的？

Answer 1

我认为循环应该足够快

[[df.loc[z,y] for z,y in zip(x,df.columns)] for x in l ]
Out[981]: [[5, 7, 3], [4, 2, 5]]

Answer 2

假设子列表的长度总是与列的长度相同的。

[[*df.lookup(i, df.columns)] for i in a]

[[5, 7, 3], [4, 2, 5]]