Взятие по индексам в NumPy
Чтобы взять элементы по списку номеров, используют функцию take.
Она работает с одномерным массивом и
возвращает новый ряд в порядке
переданных индексов. Индекс можно
повторять. В основной книге уже был
отбор целым массивом индексов; здесь
тот же смысл оформлен отдельной
функцией.
Возьмём из ряда четыре элементы в
порядке 3, 0, 2
и выведем результат:
import numpy as np
row = np.array([5, 10, 15, 20])
picked = np.take(row, [3, 0, 2])
print(picked) # выведет [20 5 15]
Из ряда [100, 200, 300, 400, 500]
соберите ряд по индексам [4, 1, 1, 0]
и выведите его.
Из ряда [7, 8, 9] возьмите
элементы с индексами [2, 2, 0]
и выведите форму (3,).
