Вы можете использовать следующий синтаксис для получения определенной строки из массива 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