打包函式引數

在函式中,你可以定義許多必需引數:

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)
...

但是你也可以使用解構語法打包引數,這樣你就可以使用 listdict 來分配變數。

打包引數列表

考慮一下你有一個值列表

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'