關注公眾號:「Python專欄」,後臺回覆「爬蟲書籍」,即可獲得2本Python爬蟲相關的電子書
無意間,看到這麼一道Python面試題:以下程式碼將輸出什麼?
def testFun():
temp = [lambda x : i*x for i in range(4)]
return temp
for everyLambda in testFun():
print (everyLambda(2))
複製程式碼
腦中默默一想,這還用說麼,肯定是:
0
2
4
6
複製程式碼
最後一看答案,竟然是:
6
6
6
6
複製程式碼
於是帶著懷疑的心態(其實是不服輸,不認錯),開啟編輯器,快速一敲,果然是:
懷疑了人生半天,本來還想黑,WTF Python…然後才想通是自己太生疏......
最後發現原因竟是:Python 的閉包的後期繫結導致的 late binding。
這意味著在閉包中的變數是在內部函式被呼叫的時候被查詢,所以當任何 testFun() 返回的函式被呼叫,i 的值是在它被呼叫時的周圍作用域中查詢。
也就是說無論哪個返回的函式被呼叫,for 迴圈都已經完成了,i 最後的值是 3,因此,每個返回的函式 testFun 的值都是 3。
因此一個等於 2 的值被傳遞進以上程式碼,它們將返回一個值 6 (比如: 3 x 2)。
究竟如何才能實現出這樣的結果呢?
0
2
4
6
複製程式碼
想了想,若能立即繫結引數,或者直接不用閉包總該行吧,用另一種方式避免 i 的改寫。
回憶了之前所學知識,最後醞釀出了四種解決方案。
第一種:建立一個閉包,通過使用預設引數立即繫結它的引數
def testFun():
temp = [lambda x, i=i: i * x for i in range(4)]
return temp
for everyLambda in testFun():
print(everyLambda(2))
複製程式碼
第二種:使用functools.partial 函式,把函式的某些引數(不管有沒有預設值)給固定住(也就是相當於設定預設值)
from functools import partial
from operator import mul
def testFun():
return [partial(mul, i) for i in range(4)]
for everyLambda in testFun():
print(everyLambda(2))
複製程式碼
第三種:優雅的寫法,直接用生成器
def testFun():
return (lambda x, i=i: i * x for i in range(4))
for everyLambda in testFun():
print(everyLambda(2))
複製程式碼
第四種:利用yield的惰性求值的思想
def testFun():
for i in range(4):
yield lambda x: i * x
for everyLambda in testFun():
print(everyLambda(2))
複製程式碼
最終執行結果:
有了解決方案後,又陷入了懷疑自己,這個題目究竟是考察的是什麼?是在考面試者閉包相關知識以及Python 的閉包的後期繫結問題麼?
若將題目改成:以下程式碼輸出的結果是(0,2,4,6)麼?如果不是,你將會怎麼做,讓它變成(0,2,4,6)?這樣會不會更有意思點呢?歡迎大家出妙招,看究竟有多少招?(哈哈哈!!!)