在 Python 中将函数输入的默认值设置为等于另一个输入

Question

考虑下面的函数，它在 Python 中不起作用，但我将用它来解释我需要做什么。

def exampleFunction(a, b, c = a):
    ...function body...

也就是说，我想为变量 c 分配与变量 a 相同的值，除非指定了替代值。上面的代码在Python中不起作用。有办法做到这一点吗？

Answer 1

def example(a, b, c=None):
    if c is None:
        c = a
    ...

关键字参数的默认值不能是变量（如果是，则在定义函数时将其转换为固定值。）通常用于将参数传递给主函数：

def main(argv=None):
    if argv is None:
        argv = sys.argv

If None可能是一个有效值，解决方案是使用 *args/**kwargs 魔法（如卡尔的答案），或使用 sentinel 对象 。执行此操作的库包括 attrs 和 棉花糖，以及在我看来，它更干净而且可能更快。

missing = object()

def example(a, b, c=missing):
    if c is missing:
        c = a
    ...

唯一使c ismissing为true的方法是c完全是您在此处创建的虚拟对象。

Answer 2

这种通用模式可能是最好且最具可读性的：

def exampleFunction(a, b, c = None):
    if c is None:
        c = a
    ...

您必须小心 None 不是 c 的有效状态。

如果你想支持“None”值，你可以这样做：

def example(a, b, *args, **kwargs):
    if 'c' in kwargs:
        c = kwargs['c']
    elif len(args) > 0:
        c = args[0]
    else:
        c = a

Answer 3

一种方法是这样的：

def foo(a, b, c=None):
    c = a if c is None else c
    # do something