1. 程式人生 > >一道神奇的Python面試題,你會嗎?

一道神奇的Python面試題,你會嗎?

開發十年,就只剩下這套架構體系了! >>>   

關注公眾號:「Python專欄」,後臺回覆「爬蟲書籍」,即可獲得2本Python爬蟲相關的電子書 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

於是帶著懷疑的心態(其實是不服輸,不認錯),開啟編輯器,快速一敲,果然是:

Python程式設計

懷疑了人生半天,本來還想黑,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程式設計 python程式設計 python程式設計 python程式設計

有了解決方案後,又陷入了懷疑自己,這個題目究竟是考察的是什麼?是在考面試者閉包相關知識以及Python 的閉包的後期繫結問題麼?

若將題目改成:以下程式碼輸出的結果是(0,2,4,6)麼?如果不是,你將會怎麼做,讓它變成(0,2,4,6)?這樣會不會更有意思點呢?歡迎大家出妙招,看究竟有多少招?(