numpy.expand_dims(a, axis) [source]

Expande la forma de una matriz.

Inserte un nuevo eje que aparecerá en el axis posición en la forma de matriz expandida.

Parámetros
aarray_like

Matriz de entrada.

axisint or tuple of ints

Posición en los ejes expandidos donde se coloca el nuevo eje (o ejes).

En desuso desde la versión 1.13.0: Pasando un eje donde axis > a.ndim será tratado como axis == a.ndimy pasando axis < -a.ndim - 1 será tratado como axis == 0. Este comportamiento está desaprobado.

Modificado en la versión 1.18.0: Ahora se admite una tupla de ejes. Los ejes fuera de rango como se describió anteriormente ahora están prohibidos y aumentan un AxisError.

Devoluciones
resultndarray

Vista de a con el número de dimensiones aumentado.

Ver también

squeeze

La operación inversa, eliminando las dimensiones singleton

reshape

Insertar, eliminar y combinar dimensiones y cambiar el tamaño de las existentes

doc.indexing, atleast_1d, atleast_2d, atleast_3d

Ejemplos de

>>> x = np.array([1, 2])
>>> x.shape
(2,)

Lo siguiente es equivalente a x[np.newaxis, :] o x[np.newaxis]:

>>> y = np.expand_dims(x, axis=0)
>>> y
array([[1, 2]])
>>> y.shape
(1, 2)

Lo siguiente es equivalente a x[:, np.newaxis]:

>>> y = np.expand_dims(x, axis=1)
>>> y
array([[1],
       [2]])
>>> y.shape
(2, 1)

axis también puede ser una tupla:

>>> y = np.expand_dims(x, axis=(0, 1))
>>> y
array([[[1, 2]]])
>>> y = np.expand_dims(x, axis=(2, 0))
>>> y
array([[[1],
        [2]]])

Tenga en cuenta que algunos ejemplos pueden usar None en lugar de np.newaxis. Estos son los mismos objetos:

>>> np.newaxis is None
True