Skip to main content
 首页 » 编程设计

Python 列表作为列表索引

2025年05月04日16over140

假设我们有两个列表

list_A = [1,3,4,54,3,5,6,2,6,77,73,39] 
list_B = [0,3,2,8] 

我想访问 list_A 的元素,这些元素将 list_B 中的值作为它们的索引(不使用循环)。

执行后,上述情况的结果应该如下:

[1, 54, 4, 6] 

是否有任何简单的方法可以做到这一点而无需使用 for 循环(在代码中显式调用它)?

请您参考如下方法:

一切都会在内部使用循环*,但它不一定像您想象的那么复杂。您可以尝试列表理解:

[list_A[i] for i in list_B] 

或者,您可以使用 operator.itemgetter :

list(operator.itemgetter(*list_B)(list_A)) 

>>> import operator 
>>> list_A = [1,3,4,54,3,5,6,2,6,77,73,39] 
>>> list_B = [0,3,2,8] 
>>> [list_A[i] for i in list_B] 
[1, 54, 4, 6] 
>>> list(operator.itemgetter(*list_B)(list_A)) 
[1, 54, 4, 6] 

* 好的!也许您不需要带递归的循环,但我认为对于这样的事情来说它肯定是过大了。