知道其索引的多个列表元素

Question

我需要从给定列表中选择一些元素，知道它们的索引。假设我想创建一个新列表，其中包含索引为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] for i in b]

有没有更好的方法呢？像c = a [b]之类的东西？

Answer 1

你可以使用operator.itemgetter：from operator import itemgetter a = [-2, 1, 5, 3, 8, 5, 6]b = [1, 2, 5]print(itemgetter(*b)(a))# Result:(1, 5, 5)或者你可以使用numpy：import numpy as npa = np.array([-2, 1, 5, 3, 8, 5, 6])b = [1, 2, 5]print(list(a[b]))# Result:[1, 5, 5]但实际上，您当前的解决方案很好。它可能是所有这些中最好的。

Answer 2

备择方案：>>> map(a.__getitem__, b)[1, 5, 5]>>> import operator>>> operator.itemgetter(*b)(a)(1, 5, 5)

Answer 3

另一种解决方案可能是通过熊猫系列：import pandas as pda = pd.Series([-2, 1, 5, 3, 8, 5, 6])b = [1, 2, 5]c = a[b]然后，您可以根据需要将c转换回列表：c = list(c)