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