简体   繁体   中英

Transform square matrix into 1D array with NumPy

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.

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.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM