乐趣区

一道神奇的Python面试题,你会吗?

关注公众号:「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)?这样会不会更有意思点呢?欢迎大家出妙招,看究竟有多少招?(哈哈哈!!!)

退出移动版