Pythonic способ вырезать определенные элементы из массива NumPy

Question

У меня есть список Python (массив numpy) и другой список, который содержит индексы для расположения значений из первого массива, который я хочу сохранить.

Есть ли способ Pythonic для этого?Я знаю numpy.delete, но я хочу сохранить элементы, а не удалять их.

Answer 1

Почему бы вам не использовать просто c=a[b], так как это способ Python для получения значений из массива a.

Answer 2

Самый питонический путь, вероятно, также самый простой:

a = np.array([2,5,6,3,6,3,45,6])
b = [0,3,4,7] # indices that you need to keep
c = a[b]

или, если вам больше не нужно a:

a = a[b]

Answer 3

Вы можете создать новый список со значениями, которые хотите сохранить.

a = np.array([2,5,6,3,6,3,45,6])
b = [0,3,4,7] #indices that yo need to keep
c = [a[i] for i in b]