1毛萌、函數(shù)作為返回值
高階函數(shù)除了可以接受函數(shù)作為參數(shù)外坯台,還可以把函數(shù)作為結(jié)果值返回兼都。
如下所示:
def lazy_sum(*args): def sum(): ax = 0 for x in args: ax = x + ax return ax return sum L = [1,3,4,5,6] f = lazy_sum(*L)#這里定義的是可變參數(shù)赢底,故L前必須加* print(f) print(f())
結(jié)果:
<function lazy_sum.<locals>.sum at 0x000001AC82150730> 19 [Finished in 0.1s]
在這個例子中殴蓬,我們在函數(shù)lazy_sum中又定義了函數(shù)sum,并且狂窑,內(nèi)部函數(shù)sum可以引用外部函數(shù)lazy_sum的參數(shù)和局部變量媳板,當(dāng)lazy_sum返回函數(shù)sum時,相關(guān)參數(shù)和變量都保存在返回的函數(shù)中泉哈,這種稱為“閉包(Closure)”的程序結(jié)構(gòu)擁有極大的威力蛉幸。
請?jiān)僮⒁庖稽c(diǎn),當(dāng)我們調(diào)用lazy_sum()時丛晦,每次調(diào)用都會返回一個新的函數(shù)奕纫,即使傳入相同的參數(shù):
f1 = lazy_sum(*L) f2 = lazy_sum(*L) print(f1 == f2)
結(jié)果:
False [Finished in 0.1s]
f1()和f2()的調(diào)用結(jié)果互不影響。
2烫沙、閉包(closure)
注意到返回的函數(shù)在其定義內(nèi)部引用了局部變量args匹层,所以,當(dāng)一個函數(shù)返回了一個函數(shù)后锌蓄,其內(nèi)部的局部變量還被新函數(shù)引用又固,所以仲器,閉包用起來簡單,實(shí)現(xiàn)起來可不容易仰冠。
另一個需要注意的問題是乏冀,返回的函數(shù)并沒有立刻執(zhí)行,而是直到調(diào)用了f()才執(zhí)行洋只。我們來看一個例子:
def count(): fs =[] for i in range(1,4): def f(): return i * i fs.append(f) return fs f1, f2, f3 = count() print(f1(), f2(), f3())
在上面的例子中辆沦,每次循環(huán),都創(chuàng)建了一個新的函數(shù)识虚,然后肢扯,把創(chuàng)建的3個函數(shù)都返回了。
你可能認(rèn)為調(diào)用f1()担锤,f2()和f3()結(jié)果應(yīng)該是1蔚晨,4,9肛循,但實(shí)際結(jié)果是:
9 9 9 [Finished in 0.2s]
全部都是9铭腕!原因就在于返回的函數(shù)引用了變量i,但它并非立刻執(zhí)行多糠。等到3個函數(shù)都返回時累舷,它們所引用的變量i已經(jīng)變成了3,因此最終結(jié)果為9夹孔。
返回閉包時牢記的一點(diǎn)就是:返回函數(shù)不要引用任何循環(huán)變量被盈,或者后續(xù)會發(fā)生變化的變量。
如果一定要引用循環(huán)變量怎么辦搭伤?方法是再創(chuàng)建一個函數(shù)只怎,用該函數(shù)的參數(shù)綁定循環(huán)變量當(dāng)前的值,無論該循環(huán)變量后續(xù)如何更改怜俐,已綁定到函數(shù)參數(shù)的值不變:
def count1(): def f(j): def g(): return j*j return g fs = [] for i in range(1,4): fs.append(f(i))#f(i)立刻被執(zhí)行身堡,因此i的當(dāng)前值被傳入f() return fs f1,f2,f3=count1() print(f1(), f2(),f3())
結(jié)果:
1 4 9 [Finished in 0.1s]
缺點(diǎn)是代碼較長,可利用lambda函數(shù)縮短代碼佑菩。