Python为后续函数调用设置参数值

Question

Python 中是否有选项可以为某些函数定义参数值，以便在调用上述函数时使用这些值？

在 TF Slim 中是这样的： https : //www.tensorflow.org/api_docs/python/tf/contrib/framework/arg_scope

例子：

set_default_params(functions=[func_1, func_2], params=[param_1=value_1, param_2=value_2, param_3=value_3])

在此之后，当我调用具有默认参数param_1=0和param_2=None func_1() ，它实际上会得到param_1=value_1和param_2=value_2而无需在调用func_1()时手动设置它。

Answer 1

您可以使用functools.partial来设置默认参数。 这部分地将参数应用于给定的函数。

import functools

def func(param, *args, **kwargs):
    print('param:', param)
    print('args:', *args)
    print('kwargs:', kwargs)

g = functools.partial(func, 21)      # sets default value for param to 21
g('arg1', 'arg2', 'arg3', kw='kwarg1')

输出：

param: 21
args: arg1 arg2 arg3
kwargs: {'kw': 'kwarg1'}

请注意，从functools.partial返回了一个新函数，并且默认参数没有像您的set_default_params调用所暗示的那样就地设置。

Answer 2

如果您希望将来能够设置参数，您可以使用某种装饰器来完成。

def kind_of_decorator(func, params):
    def wrapper(*args, **kwargs):
        return func(*args, **dict(kwargs, **params))
    return wrapper

def foo(x, y=2):
    return x + y

print(foo(1, 5)) # prints 6
print(foo(3))    # prints 5

new_foo = kind_of_decorator(foo, {'y': 0})

print(new_foo(1, y=5))  # prints 6
print(new_foo(3))       # prints 3
#print(new_foo(1, 5))   # an error "TypeError: bar() got multiple values for argument 'y'" be careful there. Now only kwargs are allowed for `y`

Answer 3

也许不是那种形式，但如果你用def function(arg={})定义一个函数，那么arg参数的默认值将在调用之间保持不变，而不是被设置为一个新对象。 这意味着您可以使用：

def function1(arg1, arg2=None, arg3=None, args={"arg2": 42, "arg3": 43}):
    if arg2 is not None:
        args["arg2"] = arg2
    if arg3 is not None:
        args["arg3"] = arg3
    arg2, arg3 = args["arg2"], args["arg3"]
    print(arg1, arg2, arg3)

function1(1)  # prints 1 42 43; using default values for arg2 and arg3
function1(1, 2) # prints 1 2 43; replace arg2 but keep arg3
function1(1, arg3=3) # prints 1 2 3; replace arg3 but keep arg2 from previous call
function1(0)  # prints 0 2 3; both arg2 and arg3 persist from previous calls.