python - 两个不同维度数组的反向相加

Question

我有两个不同尺寸的数组如下

a=np.array([[1,2,3],[1,2,3],[1,2,3],[4,5,6],[7,8,9]])

b=np.array([[4,5,6],[5,8,9]])

我怎样才能得到c，让

c[-1]=a[-1]+b[-1],   c[-2]= a[-2]+b[-2]

即： c=[[8,10,12],[12,16,18]] 我试过如下，但结果不对。

import numpy as np
c=list()
a=np.array([[1,2,3],[1,2,3],[1,2,3],[4,5,6],[7,8,9]])
b=np.array([[4,5,6],[5,8,9]])
for i in range(len(b)):
    b= a[-(len(b)+1-i)] + b
    c.append(b)
    print(c)

标签： pythonarrays