numpy.expand_dims(a, axis)
[source]
Expand the shape of an array.
Insert a new axis that will appear at the axis
position in the expanded array shape.
Note
Previous to NumPy 1.13.0, neither axis < a.ndim  1
nor axis > a.ndim
raised errors or put the new axis where documented. Those axis values are now deprecated and will raise an AxisError in the future.
Parameters: 


Returns: 

See also
>>> x = np.array([1,2]) >>> x.shape (2,)
The following is equivalent to x[np.newaxis,:]
or x[np.newaxis]
:
>>> y = np.expand_dims(x, axis=0) >>> y array([[1, 2]]) >>> y.shape (1, 2)
>>> y = np.expand_dims(x, axis=1) # Equivalent to x[:,np.newaxis] >>> y array([[1], [2]]) >>> y.shape (2, 1)
Note that some examples may use None
instead of np.newaxis
. These are the same objects:
>>> np.newaxis is None True
© 2005–2019 NumPy Developers
Licensed under the 3clause BSD License.
https://docs.scipy.org/doc/numpy1.17.0/reference/generated/numpy.expand_dims.html