Agregar columna en NumPy

Este tutorial presentará el método para agregar una columna a un array NumPy en Python.

Agregar columna a un array NumPy con la función numpy.append()

La función numpy.append() se puede usar para agregar una columna adicional a un array numpy existente. La función numpy.append() toma tres parámetros, el array preexistente, los nuevos valores que se agregarán y el axis mediante el cual queremos agregar los nuevos valores a el array preexistente. Podemos especificar el parámetro axis en 1 si queremos agregar nuevas columnas a el array preexistente. La función numpy.append() devuelve una nueva matriz en la que los nuevos valores se añaden al final del array preexistente. Vea el siguiente ejemplo 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

Producción :

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

En el código anterior, primero inicializamos nuestra primer array array con la función numpy.array() en Python. Luego inicializamos una nuevo array, array2, que contiene una columna extra, con la función numpy.append() en Python.