Чтобы получить определенную строку элементов, обратитесь к массиву numpy со всеми конкретными значениями индекса для других измерений и для строки элементов, которую вы хотите получить. Это частный случай нарезки массива в Python.
Например, представьте, что у нас есть трехмерный массив numpy формы (m, n, p). И мы хотим получить ряд элементов в i- м элементе по оси = 0 и k- м элементе по оси = 2. Используйте следующий синтаксис, чтобы получить эту желаемую строку элементов.
row = ndarray[i, :, k]
Пример 1
В следующем примере мы инициализируем трехмерный массив и получим доступ к определенной строке элементов, представленных с индексом = 0 по оси = 0 и с индексом = 1 по оси = 2.
import numpy as np
#initialize an array
arr = np.array([[[11, 11, 9, 9],
[11, 0, 2, 0]],
[[10, 14, 9, 14],
[0, 1, 11, 11]]])
# print shape of array
print('Array Shape: ',arr.shape)
# get the desired row
row = arr[0, :, 1]
print('Desired Row of Elements: ', row)
Вывод:
Array Shape: (2, 2, 4) Desired Row of Elements: [11 0]
Пример 2
В следующем примере мы инициализируем 2D-массив и получим доступ к строке и столбцу, используя нарезку массива.
import numpy as np
#initialize an array
arr = np.array([[11, 11, 9, 9],
[11, 0, 2, 0]])
print('Array\n',arr)
# get index=1 along axis=0 - this means a row in 2D
row = arr[1, :]
print('arr[1, :] : ', row)
# get index=2 along axis=1 - this means a column in 2D
row = arr[:, 2]
print('arr[:, 2] : ', row)
Вывод:
Array [[11 11 9 9] [11 0 2 0]] arr[1, :] : [11 0 2 0] arr[:, 2] : [9 2]
В этом учебнике Numpy по примерам Python мы узнали, как получить доступ к строке из массивов Numpy 2D и 3D с помощью примеров программ.