访问列表的多个元素,知道它们的索引

我需要从给定的列表中选择一些元素,知道它们的索引。假设我想创建一个新的列表,其中包含一个元素,该元素的索引为1,2,5,来自给定的列表[-2,1,5,3,8,5,6]。我所做的是:

a=[-2,1,5,3,8,5,6]
b=[1,2,5]
c=[a[i]表示b中的i]

有没有更好的办法?类似于c=a[b]的东西

您可以使用操作符.itemgetter

来自操作员导入itemgetter的


a=[-2,1,5,3,8,5,6]
b=[1,2,5]
打印(项目获取者(*b)(a))
#结果:
(1, 5, 5)

或者您可以使用numpy:

将numpy导入为np
a=np.数组([-2,1,5,3,8,5,6])
b=[1,2,5]
打印(列表(a[b]))
#结果:
[1, 5, 5]

但实际上,您当前的解决方案很好。这可能是他们中最整洁的

发表评论