Element-wise addition of two list of different lengths?

前端 未结 3 497
面向向阳花
面向向阳花 2021-01-26 20:30

How to perform element-wise addition of different length lists?
Assuming \"0\" for missing elements.
Note: len(a) will always be less than or equal to len(b)

exa

3条回答
  •  执笔经年
    2021-01-26 21:17

    You can use izip_longest:

    >>> izip_longest(a,b,fillvalue=0)
    
    >>> list(_)
    [(1, 1), (2, 2), (3, 3), (0, 4), (0, 5)]
    

    Then you can do:

    >>> [sum(t) for t in izip_longest(a,b,fillvalue=0)]
    [2, 4, 6, 4, 5]
    

提交回复
热议问题