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

Замечательно! Вы успешно подписались.
Добро пожаловать обратно! Вы успешно вошли
Вы успешно подписались на кодкамп.
Срок действия вашей ссылки истек.
Ура! Проверьте свою электронную почту на наличие волшебной ссылки для входа.
Успех! Ваша платежная информация обновлена.
Ваша платежная информация не была обновлена.