在函数内部更改函数参数的 Pythonic 方式

Question

假设我在一个函数中有一个整数参数列表，我想给每个参数加 1。有没有一种干净的方法可以做到这一点，比如这个伪代码：

def foo(a, b, c):
    for argument in arguments:
        argument += 1
    ...

特别是，我想避免在代码中显式引用每个参数，而只是遍历所有参数。将函数参数仅更改为列表不是一种选择，因为 a, b, c 在代码中具有特定含义。

Answer 1

带有显式引用

clean 方法可能是使用 iterable unpacking 和映射，例如：

def foo(a,b,c):
    a,b,c = map(lambda x:x+1,(a,b,c))
    # ...

或四舍五入：

 def foo(a,b,c):
     a,b,c = map(round,(a,b,c))
     # ...

由于这仍然是显式的，而且还保留了foo的函数签名中的参数。

您当然可以使用*args 和**kwargs 并（直接）操作它们，但这可能会导致丢失有关参数a、b 和c 的含义（它们的语义）的信息。

使用装饰器（将函数应用于所有参数）

另一种去除参数语义的方法是使用装饰器：

def apply_func_args(func):
    def dec(f):
        def g(*args,**kwargs):
            return f(*map(func,args),**{k:func(v) for k,v in kwargs.items()})
        g.__name__ = f.__name__
        g.__doc__ = f.__doc__
        return g
    return dec

然后你可以把装饰器放在foo上，比如：

@apply_func_args(round)
def foo(a,b,c):
    # ...

现在如果您调用foo(1.4,1.3,1.9)，foo 将获得(a,b,c) == (1,1,2)。例如，如果你想printa、b和c：

>>> def apply_func_args(func):
...     def dec(f):
...         def g(*args,**kwargs):
...             return f(*map(func,args),**{k:func(v) for k,v in kwargs.items()})
...         g.__name__ = f.__name__
...         g.__doc__ = f.__doc__
...         return g
...     return dec
... 
>>> @apply_func_args(round)
... def foo(a,b,c):
...     print((a,b,c))
... 
>>> foo(1.4,1.3,1.9)
(1, 1, 2)
>>> foo(1.4,2.5,c=0)
(1, 2, 0)
>>> foo(b=1.4,a=2.5,c=0)
(2, 1, 0)

因此，我们在这里定义了一种简洁的方法，将函数 (round) 应用于foo 的所有 命名和未命名参数。因此，如果我们调用foo，foo 总是会得到四舍五入的值。

Answer 2

def foo(a, b, c):
    return call_other(round(a),round(b),round(c))

或

def foo(a, b, c):
    return call_other(a+1,b+1,c+1)

我想如果我理解你想要做什么......（通过阅读 cmets）

Answer 3

这是一个没有副作用的例子：

a = [1, 2, 3, 4]
b = [5, 6, 7, 8]
c = [9, 8, 7, 6]
def foo(*args):
    modified = list()
    for idx, arg in enumerate(args):
        modified.append([x + 1 for x in arg])
    print modified
    # do something else

# In [17]: foo(a, b, c)
# [[2, 3, 4, 5], [6, 7, 8, 9], [10, 9, 8, 7]]

通过此实现，您可以输入任意数量的数组。