如何在Python中压缩数组但保留其余部分？

Question

因此，给定两个数组，两个不同的长度，我将如何组合它们，以使总和位于结果数组中。如果使用zip函数的数组长度相同，则可以执行此操作，但是我需要剩余的数组，而不仅仅是返回[4]。我在想的另一种策略是找到长度上的差异，然后追加其余部分，但我不知道这是否是最佳解决方案。我正在尝试解决二叉树合并问题，这是我的第一个解决方案。

a = [1,2,3]
b = [3]
>> zipfunction(a,b)
[4,2,3]

任何指导或建议将不胜感激。

Answer 1

使用itertools.zip_longest

>>> a = [1,2,3]
>>> b = [3]
>>> 
>>> from itertools import zip_longest
>>> list(map(sum, zip_longest(a, b, fillvalue=0)))
[4, 2, 3]