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

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

我认为for循环应该足够快

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