numpy.ndarray.ndim

атрибут

ndarray.ndim

Количество измерений массива.

Примеры

>>> x = np.array([1, 2, 3])
>>> x.ndim
1
>>> y = np.zeros((2, 3, 4))
>>> y.ndim
3