向numpy矩阵添加维数

Question

为什么下面的代码没有向矩阵中添加第三个维度？我应该如何更改代码以获得所需的内容？在

imm = np.zeros(shape=(10,10))
dims = imm.shape
if len(dims) < 3:
    imm.reshape((dims[0], dims[1], 1))
imm.shape # I want this to pring (10,10,1)

Answer 1

imm.reshape不执行就地修改。相反，它返回修改后的数组。在

因此，执行imm = imm.reshape((dims[0], dims[1], 1))，您将获得所需的输出。请考虑阅读documentation以了解更多详细信息