1. 程式人生 > >Python------偏函式(Partial function)

Python------偏函式(Partial function)

Python的functools模組提供了很多有用的功能,其中一個就是偏函式(Partial function)。要注意,這裡的偏函式和數學意義上的偏函式不一樣。

在介紹函式引數的時候,我們講到,通過設定引數的預設值,可以降低函式呼叫的難度。而偏函式也可以做到這一點。舉例如下:

int()函式可以把字串轉換為整數,當僅傳入字串時,int()函式預設按十進位制轉換:

>>> int('12345')
12345

int()函式還提供額外的base引數,預設值為10。如果傳入base引數,就可以做N進位制的轉換:

>>> int('12345', base=8)
5349
>>> int('12345', 16)
74565

假設要轉換大量的二進位制字串,每次都傳入int(x, base=2)非常麻煩,於是,我們想到,可以定義一個int2()的函式,預設把base=2傳進去:

def int2(x, base=2):
    return int(x, base)

這樣,我們轉換二進位制就非常方便了:

>>> int2('1000000')
64
>>> int2('1010101')
85

functools.partial就是幫助我們建立一個偏函式的,不需要我們自己定義int2(),可以直接使用下面的程式碼建立一個新的函式int2

>>> import functools
>>> int2 = functools.partial(int, base=2)
>>> int2('1000000')
64
>>> int2('1010101')
85

所以,簡單總結functools.partial的作用就是,把一個函式的某些引數給固定住(也就是設定預設值),返回一個新的函式,呼叫這個新函式會更簡單。

注意到上面的新的int2函式,僅僅是把base引數重新設定預設值為2,但也可以在函式呼叫時傳入其他值:

>>> int2('1000000', base=10)
1000000

最後,建立偏函式時,實際上可以接收函式物件、*args**kw這3個引數,當傳入:

int2 = functools.partial(int, base=2)

實際上固定了int()函式的關鍵字引數base,也就是:

int2('10010')

相當於:

kw = { 'base': 2 }
int('10010', **kw)

當傳入:

max2 = functools.partial(max, 10)

實際上會把10作為*args的一部分自動加到左邊,也就是:

max2(5, 6, 7)

相當於:

args = (10, 5, 6, 7)
max(*args)

結果為10

小結

當函式的引數個數太多,需要簡化時,使用functools.partial可以建立一個新的函式,這個新函式可以固定住原函式的部分引數,從而在呼叫時更簡單。