我需要从给定的列表中选择一些元素,知道它们的索引。假设我想创建一个新的列表,其中包含一个元素,该元素的索引为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]
但实际上,您当前的解决方案很好。这可能是他们中最整洁的