1. 程式人生 > >python 之 functools模組

python 之 functools模組

functools模組用於高階函式:作用於或返回其他函式的函式。一般而言,任何可呼叫物件都可以作為本模組用途的函式來處理。

官方文件已經有了詳盡的介紹,這裡就不在複述,詳情請見:官方文件 。這裡主要介紹下 partial。

partial 詳解

functools.partial返回的是一個可呼叫的partial物件,使用方法是partial(func,*args,**kw),func是必須要傳入的,而且至少需要一個args或是kw引數。

from functools import partial
def add(a, b, c):
    return a+b+c

p = partial(add, 12)
p(1, 2)
# 15
p(2, 3)
17
p(3, 4)
# 19

  建立一個功能函式,實現三個數的相加,如果其中的一個或是多個引數不變,那麼可以使用partial,例項化一個傳入了add和12引數的物件,如上圖所示,傳入兩個引數後,得到三個數的和。

partial原始碼分析:

    1.__new__方法部分

    class partial:
        """New function with partial application of the given arguments
        and keywords.
        """
        ###__SLOTS__ 只允許類有此屬性,不能動態的新增其他的屬性
        __slots__ = "func", "args", "keywords", "__dict__", "__weakref__"     
        ###__new__方法生成例項物件
        def __new__(*args, **keywords):
        ###例項化物件時傳入引數的限定,不能為空、引數的個數要大於等於2,這就解釋了至少需要一個或多個args或是kw,func是一個可呼叫的物件,是必須要傳入的
            if not args:
                raise TypeError("descriptor '__new__' of partial needs an argument")
            if len(args) < 2:
                raise TypeError("type 'partial' takes at least one argument")
            cls, func, *args = args # args=(cls,func,*args)
            if not callable(func):
                raise TypeError("the first argument must be callable")
            ### 位置引數是以元組的形式傳入的    
            args = tuple(args)

    ###hasattr這塊我也沒有咋個明白,不知道可以應用到什麼地方,從使用方法來看,傳入的函式func要有屬性或是方法,如果知道請告知我一下
        if hasattr(func, "func"):
            args = func.args + args
            tmpkw = func.keywords.copy()
            tmpkw.update(keywords)
            keywords = tmpkw
            del tmpkw
            func = func.func    
    ###建立一個例項物件本身
        self = super(partial, cls).__new__(cls)
    ###動態的新增屬性
        self.func = func
        self.args = args
        self.keywords = keywords
        return self

 上面的程式碼建立了一個例項物件(p=partial(func,*args,**kw)),並給物件本身添加了屬性。

    2. __call__方法部分

再看可呼叫的部分,partial例項化的物件是一個可呼叫的,是因為在partial中寫了__call__方法,看原始碼:

###在使用p()時會自動呼叫__call__方法	
def __call__(*args, **keywords):
    if not args:
        raise TypeError("descriptor '__call__' of partial needs an argument")
    self, *args = args
###將位置引數和關鍵字引數分別合在一起,在使用p()的時候只傳入了部分的引數,這是為了我們的方便,不重複傳入不變的引數,而在__call__方法中會將func所需的引數全部傳入	
    newkeywords = self.keywords.copy()
    newkeywords.update(keywords)
###*self.args是partial(func,*args,**kw)中的*args
    return self.func(*self.args, *args, **newkeywords)

 在使用p(*args,**keywords)的時候,就會自動的呼叫__call__方法,這就是生成的物件可呼叫的原因,self是例項化物件本身,*args、**kw是我們傳入函式func的引數,但是隻是傳入了部分引數,這也是partial的作用所在,所以還要將partial(func,*args,**kw)中的位置引數和關鍵字引數與p(*args,**keywords)一併傳入到函式func中來實現函式的功能。官網中解釋partial的功能實現相當於:

def partial(func, *args, **keywords):
    def newfunc(*fargs, **fkeywords):
        newkeywords = keywords.copy()
        newkeywords.update(fkeywords)
        return func(*args, *fargs, **newkeywords)
    newfunc.func = func
    newfunc.args = args
    newfunc.keywords = keywords
    return newfunc

 當呼叫partial函式的時候,返回的是newfunc函式物件,讓f=partial(add,12),相當了f=newfunc,所以f是一個可呼叫的物件,因此f(1,2)-->等價於newfunc(1,2)-->將引數整合傳入到func中,並返回func實現其功能。

轉自:https://blog.csdn.net/Zjack_understands/article/details/80242946