将 1d numpy 阵列重塑为 2d

Question

以下是我试图运行的代码：

import numpy as np
mask = np.zeros(1400*2100, dtype=np.uint8)
np.reshape(mask,(1400,2100), order='F')
print(mask.shape)

output 出来是： (2940000,)

我期待mask重塑为(1400,2100) 。 我的代码有什么问题？

Answer 1

np.reshape操作根本不修改mask引用，而是生成一个新数组。 如果您想对输入和结果使用相同的名称，您可以让

mask = np.reshape(mask, (1400, 2100), order='F')

或者，稍微短一点

mask = mask.reshape((1400, 2100), order='F')

Answer 2

您需要将np.reshape(mask,(1400,2100), order='F')的 output 重新分配给变量mask ：

mask=np.reshape(mask,(1400,2100), order='F')