如何最快地将矩阵压缩到NumPy中的数组/集合中？

Question

我怎样才能使矩阵像mat0 = np.array([[1,2,3],[4,5,6],[7,8,9]])展平到像arr0 = np.array([1,2,3,4,5,6,7,8,9]) ？ 还是一套？

Answer 1

使用reshape ：

>>> import numpy as np
>>> mat0 = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> mat0.reshape(-1)
array([1, 2, 3, 4, 5, 6, 7, 8, 9])

或ravel ：

>>> mat0.ravel()
array([1, 2, 3, 4, 5, 6, 7, 8, 9])