Как получить определенную строку из массива numpy (с примерами)

Вы можете использовать следующий синтаксис, чтобы получить определенную строку из массива NumPy:

 #get row in index position 2 from NumPy array
my_array[2, :]

Следующие примеры показывают, как использовать этот синтаксис на практике.

Пример 1. Получение строки из массива NumPy

Следующий код показывает, как получить определенную строку из массива NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
print (data)

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get row in index position 2
data[2,:]

array([ 9, 10, 11, 12])

Обратите внимание, что возвращается только строка с индексом 2 массива NumPy.

Пример 2. Получение нескольких строк из массива NumPy

Следующий код показывает, как получить несколько строк из массива NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get rows in index positions 0 and 2 from NumPy array
data[[0,2],:]

array([[ 1, 2, 3, 4],
       [9, 10, 11, 12]])

Пример 3. Получение строк в диапазоне из массива NumPy

Следующий код показывает, как получить строки в диапазоне из массива NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get rows in index positions 0 through 1
data[0:2, :]

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

Обратите внимание, что последнее значение диапазона (в данном случае 2) не включается в возвращаемый диапазон строк.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные операции в NumPy:

Как получить определенный столбец из массива NumPy
Как сопоставить функцию с массивом NumPy
Как добавить столбец в массив NumPy