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, la matriz preexistente, los nuevos valores que se agregarán y el axis mediante el cual queremos agregar los nuevos valores a la matriz preexistente. Podemos especificar el parámetro axis en 1 si queremos agregar nuevas columnas a la matriz preexistente. La función numpy.append() devuelve una nueva matriz en la que los nuevos valores se añaden al final de la matriz 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.