Obter Coluna de NumPy Array

Este tutorial irá apresentar o método para obter uma coluna específica de um array NumPy multidimensional em Python.

Obtenha a coluna de um array NumPy multidimensional com o método de fatiamento básico

O método de fatiamento básico funciona no mesmo princípio que o fatiamento de lista em Python. Podemos usar o método de fatiamento básico para obter uma coluna específica de um array NumPy multidimensional. O método básico de fatiamento cria uma nova visão de nosso array existente em vez de criar uma nova cópia do array. Esses novos pontos de vista na matriz existente e na memória do array original não podem ser liberados até que todas as visualizações apontando para a matriz original sejam coletadas pelo Coletor de Lixo. O exemplo de código a seguir nos mostra como obter uma coluna específica de um array NumPy multidimensional com o método básico de fatiamento em Python.

import numpy as np

array = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 0]])
print(array[:, 1])

Resultado:

[2 4 6 8 0]

No código acima, extraímos a segunda coluna do array multidimensional NumPy array com o índice de fracionamento [:,1] em Python. A primeira parte do índice é o índice das linhas. Deixamos a primeira parte em branco porque queremos selecionar todas as linhas. O operador : representa uma operação de seleção no índice. Usamos [:,1] como o índice da coluna para obter a segunda coluna de cada linha.