dimension of numpy array code example

Example 1: code for dimensions in numpy

import numpy as np

arr = np.array([1, 2, 3, 4])
arr_2D = np.array([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])

print(arr.ndim)
print(arr_2D.ndim)

Example 2: dimensions of np array python

>>> x = np.zeros((3, 5, 2), dtype=np.complex128)
>>> x.size
30
>>> np.prod(x.shape)
30