Agregar dimensión a NumPy Array

1. Agregue dimensión a NumPy Array con la función numpy.expand_dims()
2. Agregue dimensión a NumPy Array con la función numpy.newaxis en Python

Este tutorial presentará los métodos para agregar una nueva dimensión a un array NumPy en Python.

Agregue dimensión a NumPy Array con la función numpy.expand_dims()

La función numpy.expand_dims() agrega una nueva dimensión a un array NumPy. Toma el array para expandir y el nuevo eje como argumentos y devuelve una nueva matriz con dimensiones adicionales. Podemos especificar el eje a expandir dentro del parámetro axis de la función numpy.expand_dims(). Vea el siguiente ejemplo de código.

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = np.expand_dims(array, axis=0)
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

Producción :

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

En el código anterior, primero creamos un array 1D array con la función np.array() e imprimimos la forma de la array con la propiedad array.shape. Luego convertimos la array en un array 2D con la función np.expand_dims (matriz, axis=0) e imprimimos la nueva forma de la array con la propiedad matriz.shape. Al final, agregamos nuevos elementos a la array con la función np.append() e imprimimos los elementos de la array.

Agregue dimensión a NumPy Array con la función numpy.newaxis en Python

El enfoque anterior hace el trabajo y funciona bien por ahora. El único problema es que el método anterior ha quedado obsoleto y probablemente no funcionará con las versiones más nuevas de Python en el futuro. El método numpy.newaxis también se puede utilizar para lograr el mismo objetivo que el método anterior pero con un código y una complejidad aún menores. Con este método, tampoco tenemos que preocuparnos por no ser compatibles con las versiones posteriores de Python. El método numpy.newaxis agrega una nueva dimensión a nuestra matriz en Python.

import numpy as np

array = np.array([1, 2, 3])
print(array.shape)

array = array[np.newaxis]
print(array.shape)

array = np.append(array, [[4, 5, 6]], axis=0)
print(array)

Producción :

(3,)
(1, 3)
[[1 2 3]
 [4 5 6]]

Convertimos la array a un array 2D con el método array[np.newaxis] e imprimimos la nueva forma de la array con la propiedad array.shape. Al final, agregamos nuevos elementos a la array con la función np.append() e imprimimos los elementos de la array.