Numpy array dimensions

python arrays numpy 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)

Related questions

Swapping the dimensions of a numpy array

I would like to do the following: for i in dimension1: for j in dimension2: for k in dimension3: for l in dimension4: B[k,l,i,j] = A[i,j,k,l] without the use of loops. In the …

Read More
how to understand empty dimension in python numpy array?

In python numpy package, I am having trouble understanding the situation where an ndarray has the 2nd dimension being empty. Here is an example: In[1]: d2 = np.random.rand(10) In[2]: d2.shape = (-1, 1) In[3]: print d2.shape In[4]: print(d2) …

Read More
numpy array that is (n,1) and (n,)

What is the difference between a numpy array (lets say X) that has a shape of (N,1) and (N,). Aren't both of them Nx1 matrices ? The reason I ask is because sometimes computations return either one or the other.

Read More