Как получить определенный столбец из массива 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