Como obter uma coluna específica de um array numpy (com exemplos)

Você pode usar a seguinte sintaxe para obter uma coluna específica de um array NumPy:

 #get column in index position 2 from NumPy array
my_array[:, 2]

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Obtenha uma coluna do array NumPy

O código a seguir mostra como obter uma coluna específica de um array 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])

Se você deseja obter uma coluna de um array NumPy e recuperá-la como um vetor de coluna , você pode usar a seguinte sintaxe:

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

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

Exemplo 2: Obtenha várias colunas do array NumPy

O código a seguir mostra como obter várias colunas de um array 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]])

Exemplo 3: Obtenha colunas no intervalo da matriz NumPy

O código a seguir mostra como obter colunas em um intervalo de um array 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]])

Observe que o último valor do intervalo (neste caso, 3) não está incluído no intervalo da coluna retornada.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no NumPy:

Como mapear uma função para um array NumPy
Como adicionar uma coluna a um array NumPy