Преобразуйте квадратную матрицу в 1D массив с NumPy

Question

Допустим, у меня есть квадратная матрица с 20 строками и 20 столбцами. Используя NumPy, что я должен сделать, чтобы преобразовать эту матрицу в одномерный массив с одной строкой и 400 столбцами (то есть 20,20 = 400, все в одной строке)?

Пока что пробовал:

1) array = np.ravel(matrix)

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

Но когда я печатаю array, это все еще квадратная матрица.

Answer 1

Используйте метод изменения формы: array = matrix.reshape((1,400)). Это работает для типов Numpy Array и Matrix.

ОБНОВЛЕНИЕ: Как отметил Сакул, matrix.reshape(-1) является более общим с точки зрения размеров.