如何将 (r, c, 1) 矩阵重塑为 (r, c)

问题描述

给定一个形状为 (r,c,1) 的矩阵，如下所示：

[[[1],[4],[5]],[[1],[7],[6]],...
 [[9],[2],[8]]]

我想把它改造成 (r,c):

[[1,4,5],[1,7,6],[9,2,8]]

解决方法

应该这样做：

import numpy as np

np.squeeze(my_matrix)

在删除任何单位长度维度后，这将返回 my_matrix 的重塑“视图”。

替代方案：

1. my_matrix.reshape(r,c) 也返回重新塑造的视图

2. my_matrix.shape = (r,c) 改变 my_matrix“就地”的形状