Python | Numpy.expand_dims() method

With the help of Numpy.expand_dims() method, we can get the expanded dimensions of an array by using Numpy.expand_dims() method.

Syntax : Numpy.expand_dims() Return : Return the expanded array.

Example #1 : In this example we can see that using Numpy.expand_dims() method, we are able to get the expanded array using this method. Python3 1=1

# import numpy
import numpy as np
# using Numpy.expand_dims() method
gfg = np.array([1, 2])
print(gfg.shape)
gfg = np.expand_dims(gfg, axis = 0)
print(gfg.shape)

Output :

(2, ) (1, 2)

Example #2 : Python3 1=1

# import numpy
import numpy as np
# using Numpy.expand_dims() method
gfg = np.array([[1, 2], [7, 8]])
print(gfg.shape)
gfg = np.expand_dims(gfg, axis = 0)
print(gfg.shape)

Output :

(2, 2) (1, 2, 2)