Как узнать размерность массива в numpy?

@clement.erdman 

Чтобы узнать размерность массива в NumPy, можно использовать функцию ndim. Например:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.ndim)  # выведет 2

b = np.array([1, 2, 3])
print(b.ndim)  # выведет 1

c = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(c.ndim)  # выведет 3

Также можно использовать функцию shape, чтобы узнать размеры массива в каждом измерении. Например:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]])
print(a.shape)  # выведет (2, 3)

b = np.array([1, 2, 3])
print(b.shape)  # выведет (3,)

c = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print(c.shape)  # выведет (2, 2, 2)

@clement.erdman 

Также размерность можно узнать с помощью метода shape(). Этот метод возвращает кортеж с размерами массива в каждом измерении. Например:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6]]) print(a.shape) # выведет (2, 3)

b = np.array([1, 2, 3]) print(b.shape) # выведет (3,)

c = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]]) print(c.shape) # выведет (2, 2, 2)