Як отримати певний стовпець із масиву numpy (з прикладами)

Ви можете використовувати такий синтаксис, щоб отримати певний стовпець із масиву NumPy:

 #get column 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
data

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

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

array([3,7,11])

Якщо ви хочете отримати стовпець із масиву NumPy і отримати його як вектор стовпця , ви можете використати такий синтаксис:

 #get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [7],
       [11]])

Приклад 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 columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2, 4],
       [6, 8],
       [10, 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 columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

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

Зауважте, що останнє значення в діапазоні (у цьому випадку 3) не включено в діапазон стовпців, що повертається.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в NumPy:

Як зіставити функцію з масивом NumPy
Як додати стовпець до масиву NumPy