Search code examples
pythonarraysnumpysize

size of NumPy array

Is there an equivalent to the MATLAB size() command in Numpy?

In MATLAB,

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

In Python,

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

  • This is called the "shape" in NumPy, and can be requested via the .shape attribute:

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

    If you prefer a function, you could also use numpy.shape(a).