python - 使用 NumPy 将方阵转换为一维数组

Question

假设我有一个 20 行 20 列的方阵。使用 NumPy，我应该如何将这个矩阵转换为单行 400 列的一维数组（即 20.20 = 400，都在一行中）？

到目前为止，我已经尝试过：

1)array = np.ravel(matrix)

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

但是当我打印时array，它仍然是一个方阵。

标签： pythonarraysnumpymatrix

Answer 1

使用重塑方法： array = matrix.reshape((1,400))。这适用于 Numpy Array 和 Matrix 类型。

更新：正如 sacul 指出的，matrix.reshape(-1)在尺寸方面更普遍。