[英]getting elements in a list according to list of indices in python
我有一个索引列表,例如:
b=[0,2]
和元素列表:
a = ['elem0','elem1','elem2']
我需要一个由 a 中的元素和b
中a
索引组成的列表
(在本例中: ['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.