如何添加不同形状的numpy数组？

Question

我有两个numpy数组。 一个是2*2阵列。你知道吗

a = [[1,2],[3,4]]

另一个是2*2*4阵列。你知道吗

b = [[[0,0,0,0],[0,0,0,0]],[[0,0,0,0],[0,0,0,0]]]

我想把它们加起来，得到一个2*2*4的数组，c

c = [[[1,0,0,0],[2,0,0,0]],[[3,0,0,0],[4,0,0,0]]]

什么是正确的核子方法？你知道吗

编辑：这似乎是工作

  b[:,:,:1]+=a[:,:,np.newaxis]

Answer 1

不确定你是否能一步一步地求和。这里有两个步骤：

c = b.copy()
c[...,0] += a