Question

Is there a standard practice for representing vectors as 1d or 2d ndarrays in NumPy? I’m moving from MATLAB which represents vectors as 2d arrays.

Answer 1

In my experience, 1D is the norm in numpy for vectors. The only good reason to keep a vector of n elements as a 2D array of shape (1, n) or (n, 1) is in a linear algebra context, where you wanted to keep row and column vectors differentiated. As EitanT hinted on his now deleted answer, you would probably then want to use numpy’s matrix type, which keeps 2D shape of returns except for single element access, e.g if a has shape (m, n) then a[0] has shape (n,) for type ndarray, but shape (1, n) for type matrix, although a[0, 0] returns a scalar in both cases.

If you stick with 1D vector of shape (n,), you can reshape on the fly for specific operations requiring the 2D shape:

a.reshape(-1, 1) # shape (n, 1)
a[:, None] # shape (n, 1)
a.reshape(1, -1) # shape (1, n)
a[None, :] # shape (1, n)

Numpy will automatically reshape your 1D vectors to shape (1, n) when broadcasting it for an operation with a 2D array involved.