如何在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]