Python學習筆記之偏函數

標籤：python   偏函數   

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。

Python學習筆記之偏函數