python主函数参数_Python可选函数参数默认为另一个参数的值

I want to define a function with some optional arguments, let's say A (mandatory) and B (optional). When B is not given, I want it to take the same value as A. How could I do that?

I have tried this, but it doesn't work (name 'B' is not defined):

def foo(A, B=A):

do_something()

I understand that the values of the arguments are not assigned before the body of the function.

解决方案

You shall do this inside of your function.

Taking your original function:

def foo(A, B=A):

do_something()

try something like:

def foo(A, B=None):

if B is None:

B = A

do_something()

Important thing is, that function default values for function arguments are given at the time, the function is defined.

When you call the function with some value for A, it is too late as the B default value was already assigned and lives in function definition.