如何在python和numpy中将两个数组合并为一个？

Question

我想将两个数组(a,b)（每个shape = (30,192,192,1)）合并为一个（输入）像这样shape = (30,192,192,2)。
有没有人可以帮助我？

我的代码如下：

input = np.ndarray((a.shape[0], a.shape[1], a.shape[2],2))
input[:,:,:,0] = a
input[:,:,:,1] = b

但我收到此错误：

input[:,:,:,0] = a
ValueError: could not broadcast input array from shape (30,192,192,1) into shape (30,192,192)

Answer 1

您可以在之前重塑 a 和 b 数组：

a=a.reshape((30,192,192))
input[:,:,:,0] = a