python - 也使用 python 函数 arg 作为默认参数

Question

我想编写一个函数，它的默认值是函数的另一个输入。我想要这样的东西:

def pythag_thm(a, b=a):
    return (a**2 + b**2)**.5

但我得到一个NameError: name 'a' is not defined。我意识到我可以做类似的事情:

def pythag_thm(a, b=False):
     if b==False:
         return (a**2 + a**2)**.5
     else:
         return (a**2 + b**2)**.5

但这似乎不是很“pythonic”。有没有一种干净的方法可以做到这一点？我只是以错误的方式思考这个问题吗？

Answer 1

人们通常做的是

def pythag_thm(a, b=None):
    if b is None:
        b = a

    return (a ** 2 + b ** 2) ** .5