Transform square matrix into 1D array with NumPy

Question

Let's say I have a square matrix with 20 lines and 20 columns. Using NumPy, what should I do to transform this matrix into a 1D array with a single line and 400 columns (that is, 20.20 = 400, all in one line)?

So far, I've tried:

1) array = np.ravel(matrix)

2) array = np.squeeze(np.asarray(matrix))

But when I print array, it's still a square matrix.

Also try np.reshape() it should work(more detail [docs.scipy.org/doc/numpy/reference/generated/…). — Vage Egiazarian
– Vage Egiazarian, Commented Aug 28, 2018 at 15:53
If you're using a numpy matrix, then switch to an ordinary ndarray. The matrix object is deprecated. array = matrix.ravel() should work if matrix is an ndarray. — user2653663
– user2653663, Commented Aug 28, 2018 at 15:56
If ravel doesn't work, then there's something unusual about matrix. Tell us its shape, dtype, and whether it is np.matrix or not. — hpaulj
– hpaulj, Commented Aug 28, 2018 at 15:57

smp55 · Accepted Answer · 2018-08-28 15:54:58Z

3

Use the reshape method: array = matrix.reshape((1,400)). This works for both Numpy Array and Matrix types.

UPDATE: As sacul noted, matrix.reshape(-1) is more general in terms of dimensions.

answered Aug 28, 2018 at 15:54

smp55

4033 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Transform square matrix into 1D array with NumPy

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related