Як отримати певний рядок із масиву 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