问题描述
我需要相互添加 2 个可变大小的暗淡数组。有很多方法可以做到这一点!
典型大小是几千乘几百(因为这可能会影响缩放!)。需要进行数十万次这些添加。第一维在我的情况下保证是相同的,但子数组是可变长度的。
工作较小的例子:
a = np.ones(shape=(20,20))
b = np.ones(shape=(20,18))
c = a+b # Expected error
b.resize(a.shape)
c = b+c # This works!
有没有更快的方法来做到这一点?我对上述其他 Pythonic 解决方案感兴趣,但也对无论复杂性如何都可能真正最快的解决方案感兴趣(这种项目的速度主要是速度,但还不如用 C 编写)。
解决方法
如果你不介意改变a
,你可以这样做
a[:,:b.shape[1]] += b
我怀疑这比填充 b
然后添加要快,但像这样的事情需要用真实数据进行测试。