Numpy array dimensions

morgan freeman picture morgan freeman · Jun 17, 2010 · Viewed 829.3k times · Source

I'm currently trying to learn Numpy and Python. Given the following array:

import numpy as np
a = np.array([[1,2],[1,2]])

Is there a function that returns the dimensions of a (e.g.a is a 2 by 2 array)?

size() returns 4 and that doesn't help very much.

Answer

Felix Kling picture Felix Kling · Jun 17, 2010

It is .shape:

ndarray.shape
Tuple of array dimensions.

Thus:

>>> a.shape
(2, 2)