Python Program, чтобы получить определенную строку из Numpy Array

Автор оригинала: Python Examples.

Получить ряд от Numpy Array

Чтобы получить конкретный ряд элементов, доступа к массиву Numpy со всеми конкретными значениями индекса для других измерений и : Для ряда элементов вы хотели бы получить. Это особый случай нарезки массива в Python.

Например, считайте, что у нас есть 3D Numpy Array формы (M, N, P). И мы хотели бы получить ряд элементов в I ^TH Элемент вдоль, и k ^TH Элемент вместе. Используйте следующий синтаксис, чтобы получить эту нужную строку элементов.

row = ndarray[i, :, k]

Пример 1: Доступ к определенному ряду элементов

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

Python Program

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 Numpy Array

В следующем примере мы будем инициализировать 2D Array и доступ к ряд и столбцу, используя нарезку массива.

Python Program

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-массивов, с помощью хорошо подробных примеров программ.