Как извлечь элементы из списка, используя индексы в Python?

Если у вас есть список в 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])

Ответы

Ответ 1

Возможно, используйте это:

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

Ответ 2

Try

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

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

Ответ 3

Я думаю, вы ищете это:

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

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

Ответ 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])

Ответ 5

Проверены флажки:

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