Python的functiools模塊提供了很多有用的功能,其中一個(gè)就是偏函數(shù)(Partial function)。要注意,這里的偏函數(shù)和數(shù)學(xué)意義上的偏函數(shù)不一樣。
在介紹函數(shù)參數(shù)的時(shí)候,我們講到,通過(guò)設(shè)定參數(shù)的默認(rèn)值,可以降低函數(shù)調(diào)用的難度。而偏函數(shù)也可以做到這一點(diǎn)。舉例:
int()函數(shù)可以把字符串轉(zhuǎn)換為整數(shù),當(dāng)僅傳入字符串時(shí),int()函數(shù)默認(rèn)按十進(jìn)制轉(zhuǎn)換:
>>> int('12345')
12345
但int()函數(shù)還提供額外的base參數(shù),默認(rèn)值為10。如果傳入base參數(shù),就可以做N進(jìn)制的轉(zhuǎn)換:
>>> int('12345', base = 8)
5349
>>> int ('12345',16)
74565
假設(shè)要轉(zhuǎn)換大量的二進(jìn)制字符串,每次都傳入int(x, base=2)非常麻煩,于是,我們想到,可以定義一個(gè)int2()的函數(shù),默認(rèn)把base=2傳進(jìn)去:
def int2(x, base=2):
return int (x, base)
這樣,我們轉(zhuǎn)換二進(jìn)制就非常方便了:
>>> int2 ('1000000')
64
>>> int2('1010101')
85
functools.partial就是幫助我們創(chuàng)建一個(gè)偏函數(shù)的,不需要我們自己定義int2(),可以直接使用下面的代碼創(chuàng)建一個(gè)新的函數(shù)int2:
>>> import functools
>>> int2 = functools.partial(int, base = 2)
>>> int2('1000000')
64
>>> int2('1010101')
85
所以,簡(jiǎn)單總結(jié)functools.partial的作用就是,把一個(gè)函數(shù)的某些參數(shù)給固定住(也就是設(shè)置默認(rèn)值),返回一個(gè)新的函數(shù),調(diào)用這個(gè)新函數(shù)會(huì)更簡(jiǎn)單。
注意到上面的新的int2函數(shù),僅僅是把base參數(shù)重新設(shè)定默認(rèn)值為2,但也可以在函數(shù)調(diào)用時(shí)傳入其他值:
>>> int2('1000000', base = 10)
1000000
最后,創(chuàng)建偏函數(shù)時(shí),實(shí)際上可以接受函數(shù)對(duì)象、*args和**kw這3個(gè)參數(shù),當(dāng)傳入:
int2 = functools.partial(int, base = 2)
實(shí)際上固定了int()函數(shù)的關(guān)鍵字參數(shù)base,也就是:
int2('10010')
相當(dāng)于:
kw = {'base':2}
int('10010', **kw)
當(dāng)傳入:
max2 = functools.partial(max, 10)
實(shí)際上會(huì)把10作為*args的一部分自動(dòng)加到左邊,也就是:
max2(5,6,7)
相當(dāng)于:
args = (10, 5, 6, 7)
max(*args)
10