Como adicionar uma coluna a um array numpy (com exemplos)

Você pode usar um dos seguintes métodos para adicionar uma coluna a um array NumPy:

Método 1: adicione uma coluna no final da tabela

 n.p. append (my_array, [[value1], [value2], [value3], ...], axis= 1 )

Método 2: Insira uma coluna em uma posição específica da tabela

 n.p. insert (my_array, 3 , [value1, value2, value3, ...], axis= 1 )

Os exemplos a seguir mostram como usar cada método na prática.

Exemplo 1: Adicione uma coluna no final do array NumPy

Suponha que temos o seguinte array NumPy:

 import numpy as np

#create NumPy array
my_array = np. array ([[1, 2, 3, 4], [5, 6, 7, 8]])

#view NumPy array
my_array

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

Podemos usar a seguinte sintaxe para adicionar uma coluna no final do array NumPy:

 #append column to end of NumPy array
new_array = np. append (my_array, [[10], [13]], axis= 1 )

#view updated array
new_array

array([[ 1, 2, 3, 4, 10],
       [5, 6, 7, 8, 13]])

Exemplo 2: Insira uma coluna em uma posição específica do array NumPy

Suponha que temos o seguinte array NumPy:

 import numpy as np

#create NumPy array
my_array = np. array ([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

#view NumPy array
my_array

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

Podemos usar a seguinte sintaxe para inserir uma nova coluna antes da coluna na posição 2 do índice do array NumPy:

 #insert new column before column in index position 2
new_array = np. insert (my_array, 2 , [10, 13, 19], axis= 1 )

#view updated array
new_array

array([[ 1, 2, 10, 3],
       [4, 5, 13, 6],
       [7, 8, 19, 9]])

Observe que a nova coluna de valores foi inserida antes da coluna na posição 2 do índice.

Recursos adicionais

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

Como adicionar um array Numpy a um DataFrame do Pandas
Como converter Pandas DataFrame em matriz NumPy