numpy print dimensions 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: numpy get array size

myArray.shape # Returns the number of rows, columns etc. (depends on dimensions how many numbers you get)

print(a_1d.shape)
# (3,)

print(type(a_1d.shape))
# <class 'tuple'>

print(a_2d.shape)
# (3, 4)

print(a_3d.shape)
# (2, 3, 4)