размер массива NumPy

Существует ли эквивалент MATLAB

 size()

команда в Numpy?

В MATLAB,

>>> a = zeros(2,5)
 0 0 0 0 0
 0 0 0 0 0
>>> size(a)
 2 5

В Python,

>>> a = zeros((2,5))
>>> 
array([[ 0.,  0.,  0.,  0.,  0.],
       [ 0.,  0.,  0.,  0.,  0.]])

>>> ?????

Ответы

Ответ 1

Это называется "форма" в NumPy и может быть запрошено через атрибут .shape:

>>> a = zeros((2, 5))
>>> a.shape
(2, 5)

Если вы предпочитаете функцию, вы также можете использовать numpy.shape(a).