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

Пользователь

от clement.erdman , в категории: Python , месяц назад

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

Facebook Vk Ok Twitter LinkedIn Telegram Whatsapp

1 ответ

Пользователь

от felix , месяц назад

@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)