添加特定值作为numpy数组的新维

Question

读取图像后，我得到一个形状为（224,224,3）的numpy数组。但是，我想将其转换为（4,224,224,3）的形状。

我想重复相同的值。

我正在尝试像下面所示的那样添加内容，但是它不起作用。

np.append(image,[[[4]]],axis=1)

相反，它引发以下错误

ValueError: all the input arrays must have same number of dimensions

我希望输出形状为（4,224,224,3）

您能指导我如何做吗？

Answer 1

您可以使用np.repeat将轴设置为0：

out = np.repeat([image], 4, axis=0)
out.shape
# (4, 224, 224, 3)