python - 如何将不定数量的列表添加在一起并输出单个列表？

Question

这个问题已经有答案了:

What is the best way to iterate over multiple lists at once? [duplicate] (2 个回答)

已关闭 4 年前。

输入将是列表(列表的数量是不确定的)，该函数应该迭代所有索引，并将所有输入列表的相同索引的列表的每个值以数学方式添加到一起。输出将是一个包含所有添加值的列表

例如:列表 = [1,2,3] 列表b = [2,3,5] 列表c = [-3,2,1]输出列表 = [0,7,9]我的下面的函数只能将 2 个列表添加在一起，我不希望限制列表的数量。我怎么做？预先非常感谢您

def listadd(a,b):
    counter = 0
    list = []
    while counter < len(a):
        list.append(a[counter]+b[counter])
        counter += 1
    return list

Answer 1

您可以使用map与 zip :

def listadd_new(*lsts):
    return list(map(sum, zip(*lsts)))

assert listadd([1, 2, 3], [4, 5, 6]) == listadd_new([1, 2, 3], [4, 5, 6])