前几天在知乎上看见一道Python题 如下:
def testrun():
temmp=[lambda x: x*i for i in range(4)]
return temmp
for everyLambda in testrun():
print(everyLambda(2))
自己想了一下,自以为答案是 0,2,4,6 但是在编译器上运行 得到的答案是 6,6,6,6
主要原因是因为Python中的闭包问题,变量是在函数被调用的时候在其作用范围中找到变量,所以当任何testrun进行调用时,i的值是在它被调用时的周围作用
域中查找,但是无论返回哪个函数,for循环已经执行完毕了,i最后的值是3,因此返回值为6 闭包问题如下:
python闭包问题
解决方案如下:
# 1 创建一个闭包,通过使用默认参数立即绑定它的参数
def testrun1():
temp=[lambda x,i=i:i*x for i in range(4)]
return temp
for everyLambda in testrun1():
print(everyLambda(2))
2使用functools.partial函数把函数中的参数固定住
from functools import partial
from operator import mul
# mul函数 mul(x,y) return x*y
def testrun2():
return [partial(mul,i) for i in range(4)]
for everyLambda in testrun2():
print(everyLambda(2))
3直接使用生成器
def testrun3():
return (lambda x,i=i: x*i for i in range(4))
for everyLambda in testrun3():
print(everyLambda(2))
4使用yiled惰性求值思想
def testrun4():
for i in range(4):
yield lambda x:i*x
for everyLambda in testrun4():
print(everyLambda(2))