python - 根据函数句柄输入正确数量的变量

Question

我有一个变量列表和一个函数对象，我想根据函数分配正确数量的变量。

def sum2(x,y):
    return x + y

def sum3(x,y,z):
    return x + y + z

varList = [1,2,3]

所以如果 f = sum2 ，我希望它调用 varList 的前 2 个元素，如果 f = sum3 ，则用函数的 3 个元素调用它。

Answer 1

如果您总是返回所有传递参数的总和，这可以在单个函数中完成。

def sum1(*args):
    return sum(args)

这只是利用位置参数，因为您似乎不需要显式设置各个值。它也比 ZdaR 提供的解决方案最灵活，因为您不需要提前知道可以接收的最大参数数量。

一些例子:

>>> print sum1(1, 2, 3)
6
>>> print sum1(1)
1
>>> print sum1(-1, 0, 6, 10)
15