Python：函数的默认值作为函数参数

Question

假设我有以下函数：

def myF(a, b):
    return a*b-2*b

假设我希望 b 的默认值为 a-1。

如果我写：

def myF(a, b=a-1):
    return a*b-2*b

我收到错误消息：

NameError: name 'a' is not defined

我可以使用下面的代码：

def myF(a, b):
    return a*b-2*b

def myDefaultF(a):
    return myF(a, a-1)

让 myF 具有默认值，但我不喜欢它。

如何避免 myDefaultF 并使 myF 为 b 使用默认值 a-1 而不会出现错误？

Answer 1

您可以执行以下操作：

def myF(a, b=None):
    if b is None:
        b = a - 1
    return a * b - 2 * b

Answer 2

如果您需要将 b 的值设为 a 的函数，但您可能需要更改该函数，则可以设置 b< /code> 为 lambda 函数，然后检查函数块中是否可调用 b 。

def myF(a, b=lambda a: a-1):
    if callable(b):
        b = b(a)
    return a * b - 2 * b

这允许您动态地为 b 设置不同的函数。

# pass b as an integer
myF(1, 1)
# returns: -1


# use default function for b
myF(4)
# returns: 6


# set b to be 2*a + 1
myF(3, lambda a: 2*a+1)
# returns: 7

Answer 3

您可以使用 try- except 子句：

def myF(a, b=None):
  try:
    return a*b-2*b
  except:
    return a*(a-1)-2*(a-1)