Есть ли способ нарезать список по его простым индексам?

Поскольку есть тег numpy:

import numpy as np
a = np.array(a)
b = np.array(b)

result = a[b]

Сравнение производительности по времени:

import numpy as np
N=100_000

a = list(range(N))
b = list(range(1, N, 3))

c = np.array(a)
d = np.array(b)

Понимание списка

%timeit [a[i] for i in b]
2.31 ms ± 456 µs per loop (mean ± std. dev. of 7 runs, 100 loops each)

Нумпи

%timeit c[d]
115 µs ± 7.57 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

Используя простой тест, мы видим, что numpy работает в 20 раз быстрее с массивом длиной 100_000, однако это некорректное сравнение, поскольку я не учитываю время на импорт библиотеки и преобразование списков в массивы.

В качестве последнего замечания: на самом деле нет необходимости также преобразовывать b в массив numpy.

Как насчет:

result = [a[i] for i in b]