Индексация массива NumPy

Доступ к элементам массива

Индексация массива эквивалентна доступу к элементам массива.
Вы можете получить элемент массива, указав его индекс.
Индексация массивов NumPy начинается с 0, что означает, что индекс первого элемента равен 0, второго элемента равен 1 и так далее.

Сначала давайте рассмотрим операции с индексами одмерного массива:

>>> import numpy as np
>>> arr = np.array([1, 2, 3, 4])
>>> print(arr[0]) # вывод первого элемента массива
1
>>> print(arr[1]) # вывод второго элемента массива
2
>>> print(arr[2] + arr[3]) # сложение третьего и четвертого элементов массива
7

Доступ к 2-D массиву

Чтобы получить элемент двумерного массива, мы можем использовать через запятую отделяемые целые числа для указания измерения и индекса.

Сначала давайте рассмотрим операции с индексами двумерного массива:

>>> import numpy as np
>>> arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
>>> print('2-й элемент во 1-м измерении: ', arr[0, 1]) # доступ ко второму элементу в первом измерении
2-й элемент во 1-м измерении: 2
>>> print('5-й элемент во 2-м измерении: ', arr[1, 4]) # доступ к пятому элементу во втором измерении:
5-й элемент во 2-м измерении: 10

Доступ к 3-D массиву

Чтобы получить доступ к элементу в 3-D массиве, мы можем использовать запятую для разделения чисел, представляющих размерность и индекс элемента.

>>> import numpy as np
>>> arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
>>> print(arr[0, 1, 2])
6

Первый номер代表着 первое измерение, которое содержит два массива:
[[1, 2, 3], [4, 5, 6]] и [[7, 8, 9], [10, 11, 12]], поскольку мы выбрали 0Таким образом, мы остались с первым массивом:[[1, 2, 3], [4, 5, 6]].
Второй номер代表着 второе измерение, которое также содержит два массива:[1, 2, 3]и [4, 5, 6]Поскольку мы выбрали 1, мы остались со вторым массивом: [4, 5, 6].
Третий номер代表着 третье измерение, которое содержит три значения: 4, 5, 6
Поскольку мы выбрали 2, в итоге мы получили третий элемент: 6

Отрицательные индексы

Использование отрицательных индексов для доступа к элементам массива с конца.

Печать последнего элемента во втором измерении:

>>> import numpy as np
>>> arr = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
>>> print('Последний элемент во втором измерении:', arr[1, -1])
Последний элемент во втором измерении: 10