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

【问题标题】：How to add differently shaped numpy arrays?如何添加不同形状的numpy数组？
【发布时间】：2011-12-26 18:23:51
【问题描述】：

我有 2 个 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]]]

正确的numpythonic方法是什么？

编辑：这似乎有效

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

【问题讨论】：

【解决方案1】：

不确定是否可以一步完成。这里分两步：

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

【讨论】：