Numpy添加两个不同大小的向量

Question

如果我有两个不同大小的numpy数组,我怎么能叠加它们.

a = numpy([0, 10, 20, 30])
b = numpy([20, 30, 40, 50, 60, 70])

添加这两个向量以生成新向量(20,40,60,80,60,70)的最简洁方法是什么？

这是我的一般性问题.对于背景,我特别应用了格林变换函数,并且需要将每个时间步骤的结果叠加到先前累积的响应中.

Answer 1

这可能就是你要找的东西

if len(a) < len(b):
    c = b.copy()
    c[:len(a)] += a
else:
    c = a.copy()
    c[:len(b)] += b

基本上你复制较长的一个,然后在原地添加较短的一个

Answer 2

如果您知道b更高的维度,那么:

>>> a.resize(b.shape)
>>> c = a+b

是你所需要的全部.