Skip to main content
 首页 » 编程设计

python之如何使用索引列表索引 Pandas 数据框

2025年05月04日13lidabo

我有一个如下所示的数据框。

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 循环,哪种方法最快?

请您参考如下方法:

我认为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]]