繁体   English   中英

根据 python 中的索引列表获取列表中的元素

[英]getting elements in a list according to list of indices in python

我有一个索引列表,例如:

b=[0,2]  

和元素列表:

a = ['elem0','elem1','elem2'] 

我需要一个由 a 中的元素和ba索引组成的列表
(在本例中: ['elem0','elem2']

使用列表理解

[a[i] for i in b]

或者:

from operator import itemgetter

b=[0,2]
a = ['elem0','elem1','elem2']

print itemgetter(*b)(a)
>>> ('elem0','elem2')

使用列表理解 map 列表索引:

b=[0,2]
a = ['elem0','elem1','elem2'] 

sublist = [a[i] for i in b]
>ipython
In [1]: b=[0,2]
In [2]: a = ['elem0','elem1','elem2']
In [3]: [a[i] for i in b]
Out[3]: ['elem0', 'elem2']

如果您不知道,请在 python 手册中查找“列表理解”。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM