python - 编写一个使用 *args 添加列表元素的函数

Question

我想编写一个函数来计算多个向量的标量和。它应该将多个列表作为参数，例如。

def add_lists(x, y, z):
   # return sum of x, y, z

x = add_lists([1, 2],[2, 1],[3, 1])
# x returns [6, 4]

只有两个参数很容易做到这一点，但是如果它们超过两个，我如何使用 *args 来做到这一点，以便我的 add_lists 将返回 [6, 4]？

Answer 1

您可以使用zip 和*args:

def add_lists(*args):
    return [sum(x) for x in zip(*args)]

x = add_lists([1, 2],[2, 1],[3, 1])
# x is [6, 4]