打包函式引數
在函式中,你可以定義許多必需引數:
def fun1(arg1, arg2, arg3):
return (arg1,arg2,arg3)
這將使函式只有在給出三個引數時才可呼叫:
fun1(1, 2, 3)
你可以使用預設值將引數定義為可選引數:
def fun2(arg1='a', arg2='b', arg3='c'):
return (arg1,arg2,arg3)
所以你可以用很多不同的方式呼叫這個函式,比如:
fun2(1) → (1,b,c)
fun2(1, 2) → (1,2,c)
fun2(arg2=2, arg3=3) → (a,2,3)
...
但是你也可以使用解構語法打包引數,這樣你就可以使用 list
或 dict
來分配變數。
打包引數列表
考慮一下你有一個值列表
l = [1,2,3]
你可以使用*
語法將值列表作為引數呼叫該函式:
fun1(*l)
# Returns: (1,2,3)
fun1(*['w', 't', 'f'])
# Returns: ('w','t','f')
但是,如果你沒有提供長度與引數數量匹配的列表:
fun1(*['oops'])
# Raises: TypeError: fun1() missing 2 required positional arguments: 'arg2' and 'arg3'
打包關鍵字引數
現在,你還可以使用字典打包引數。你可以使用**
運算子告訴 Python 將 dict
解壓縮為引數值:
d = {
'arg1': 1,
'arg2': 2,
'arg3': 3
}
fun1(**d)
# Returns: (1, 2, 3)
當函式只有位置引數(沒有預設值的那些)時,你需要包含所有預期引數的字典,並且沒有額外的引數,否則你會得到一個錯誤:
fun1(**{'arg1':1, 'arg2':2})
# Raises: TypeError: fun1() missing 1 required positional argument: 'arg3'
fun1(**{'arg1':1, 'arg2':2, 'arg3':3, 'arg4':4})
# Raises: TypeError: fun1() got an unexpected keyword argument 'arg4'
對於具有可選引數的函式,可以使用相同的方式將引數打包為字典:
fun2(**d)
# Returns: (1, 2, 3)
但是你可以省略值,因為它們將被預設值替換:
fun2(**{'arg2': 2})
# Returns: ('a', 2, 'c')
和以前一樣,你不能提供不是現有引數的額外值:
fun2(**{'arg1':1, 'arg2':2, 'arg3':3, 'arg4':4})
# Raises: TypeError: fun2() got an unexpected keyword argument 'arg4'
在實際使用中,函式可以同時具有位置和可選引數,並且它的工作原理相同:
def fun3(arg1, arg2='b', arg3='c')
return (arg1, arg2, arg3)
你可以用一個 iterable 呼叫函式:
fun3(*[1])
# Returns: (1, 'b', 'c')
fun3(*[1,2,3])
# Returns: (1, 2, 3)
或者只是一本字典:
fun3(**{'arg1':1})
# Returns: (1, 'b', 'c')
fun3(**{'arg1':1, 'arg2':2, 'arg3':3})
# Returns: (1, 2, 3)
或者你可以在同一個呼叫中使用兩個:
fun3(*[1,2], **{'arg3':3})
# Returns: (1,2,3)
請注意,你不能為同一引數提供多個值:
fun3(*[1,2], **{'arg2':42, 'arg3':3})
# Raises: TypeError: fun3() got multiple values for argument 'arg2'