Как добавить разные по форме массивы?

Question

У меня есть 2 массива.Один - это массив 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