Метод take
Метод take класса ndarray возвращает новый массив,
содержащий элементы исходного массива, расположенные по указанным индексам.
Первым параметром метод принимает список или массив индексов.
Вторым параметром можно указать ось, вдоль которой производится выборка.
Этот метод часто используется для выборки элементов по индексам
в многомерных массивах.
Синтаксис
arr.take(indices, axis=None)
Пример
Давайте выберем из массива элементы с индексами 0, 2 и 4:
import numpy as np
arr = np.array([10, 20, 30, 40, 50])
res = arr.take([0, 2, 4])
print(res)
Результат выполнения кода:
[10 30 50]
Пример
Давайте выберем элементы из двумерного массива вдоль оси 0 (по строкам):
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
res = arr.take([0, 2], axis=0)
print(res)
Результат выполнения кода:
[[1 2 3]
[7 8 9]]
Пример
Давайте выберем элементы из двумерного массива вдоль оси 1 (по столбцам):
import numpy as np
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
res = arr.take([1, 3], axis=1)
print(res)
Результат выполнения кода:
[[2 4]
[6 8]]
