如何添加不同形狀的numpy數組？

Question

我有2個numpy數組。 一個是2 * 2數組。另一個是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]]] 

什麼是正確的numpythonic方式做到這一點？

編輯：這似乎工作

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

Answer 1

不知道你是否能做到的和在一個單一的步驟。這裏分兩步：

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