Adicionar coluna em NumPy

Este tutorial irá apresentar o método para adicionar uma coluna a um array NumPy em Python.

Adicionar coluna a um array NumPy com a função numpy.append()

A função numpy.append() pode ser usada para adicionar uma coluna extra a um array numpy existente. A função numpy.append() usa três parâmetros, o array pré-existente, os novos valores a serem adicionados e o axis pelo qual queremos anexar os novos valores ao array pré-existente. Podemos especificar o parâmetro axis como 1 se quisermos adicionar novas colunas ao array pré-existente. A função numpy.append() retorna uma nova matriz na qual os novos valores são acrescentados no final do array pré-existente. Veja o seguinte exemplo de código.

import numpy as np

array = np.array([[1, 2], [3, 4], [5, 6]])
array2 = np.append(array, [[1], [1], [1]], axis=1)
array2

Resultado:

[[1 2 1]
 [3 4 1]
 [5 6 1]]

No código acima, primeiro inicializamos nosso primeiro array array com a função numpy.array() em Python. Em seguida, inicializamos um novo array, array2, contendo uma coluna extra, com a função numpy.append() em Python.