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

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循环应该足够快

response.keys.each do |field_name|
  response[field_name].each do |value|
     puts "#{field_name} => #{value}"
     User.create!(
       first_name: value['first_name'],
       last_name: value['last_name'],
       email:  value['email']
     )
  end
end

Answer 2

假定子列表的长度始终与列的长度相同

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

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