如何使用Python中的索引从列表中提取元素？

Question

如果你在python中有一个列表，并希望将索引中的元素1,2和5提取到一个新列表中，你会怎么做？

我就是这样做的，但我不是很满意

>>> a
[10, 11, 12, 13, 14, 15]
>>> [x[1] for x in enumerate(a) if x[0] in [1,2,5]]
[11, 12, 15]

更好的方法吗？

更一般地说，给定一个带索引的元组，如何使用这个元组从列表中提取相应的元素，最终重复（例如元组（1,1,2,1,5）产生[11,11， [12,11,15]）

Answer 1

也许使用这个：

[a[i] for i in (1,2,5)]
# [11, 12, 15]

Answer 2

我认为你正在寻找这个：

elements = [10, 11, 12, 13, 14, 15]
indices = (1,1,2,1,5)

result_list = [elements[i] for i in indices]    

Answer 3

尝试

numbers = range(10, 16)
indices = (1, 1, 2, 1, 5)

result = [numbers[i] for i in indices]

Answer 4

使用Numpy直接数组索引，如MATLAB，Julia，...

a = [10, 11, 12, 13, 14, 15];
s = [1, 2, 5] ;

import numpy as np
list(np.array(a)[s])
# [11, 12, 15]

更好的是，继续使用Numpy数组

a = np.array([10, 11, 12, 13, 14, 15])
a[s]
#array([11, 12, 15])

Answer 5

检查边界：

 [a[index] for index in (1,2,5,20) if 0 <= index < len(a)]
 # [11, 12, 15]