Python Numpy - Get Array Shape or Dimensions
NumPy - Array Shape or Dimensions
To get the shape or dimensions of a NumPy Array, use ndarray.shape
where ndarray is the name of the numpy array you are interested of. ndarray.shape returns a tuple with dimensions along all the axis of the numpy array.
Examples
1. Get shape of multi-dimensional NumPy array
In the following example, we have initialized a multi-dimensional numpy array. Of course, we know the shape of the array by its definition. But, we will use ndarray.shape property to get the shape of the array programmatically.
Python Program
import numpy as np
#initialize an array
arr = np.array([[[11, 11, 9, 9],
[11, 0, 2, 0]
],
[[10, 14, 9, 14],
[0, 1, 11, 11]]])
# get array shape
shape = arr.shape
print(shape)
Output
(2, 2, 4)
2. Get shape of 2D NumPy array
In the following example, we will create a 2D numpy array and find its shape. There are two rows and four columns. So, we should get a tuple of (2, 4). Let us see.
Python Program
import numpy as np
#initialize an array
arr = np.array([[11, 11, 9, 9],
[11, 0, 2, 0]])
# get array shape
shape = arr.shape
print(shape)
Output
(2, 4)
3. Get shape of 1-Dimensional NumPy array
In the following example, we take a one-dimensional numpy array, and find its shape.
Python Program
import numpy as np
#initialize an array
arr = np.array([11, 11, 9, 9])
# get array shape
shape = arr.shape
print(shape)
There are four elements, and of course, the shape should be a tuple with 4.
Output
(4,)
Summary
In this Numpy Tutorial, we learned how to get the shape of a given numpy array as a tuple.