我们在函数
lazy_sum
中又定义了函数sum
,并且,内部函数sum可以引用外部函数lazy_sum
的参数和局部变量,当lazy_sum
返回函数sum
时,相关参数和变量都保存在返回的函数中,这种称为“闭包(Closure)
”
def lazy_sum(*args):
def sum():
ax = 0
for n in args:
ax = ax + n
return ax
return sum
注意到返回的函数在其定义内部引用了局部变量args
,所以,当一个函数返回了一个函数后,其内部的局部变量还被新函数引用。
当我们调用lazy_sum()
时,返回的并不是求和结果,而是求和函数:
>>> f = lazy_sum(1, 3, 5, 7, 9)
>>> f
<function lazy_sum.<locals>.sum at 0x101c6ed90>
调用函数f时,才真正计算求和的结果:
>>> f()
25
另一个需要注意的问题是,返回的函数并没有立刻执行,而是直到调用了f()才执行。我们来看一个例子:
>>>def count():
... fs = []
... for i in range(1, 4):
... def f():
... return i*i
... fs.append(f)
... return fs
...
>>>f1, f2, f3 = count()
在上面的例子中,每次循环,都创建了一个新的函数,然后,把创建的3个函数都返回了。
你可能认为调用f1()
,f2()
和f3()
结果应该是1,4,9,
但实际结果是:
>>> f1()
9
>>> f2()
9
>>> f3()
9
如果对count()
函数进行修改:
>>>def count():
... fs = []
... for i in range(1, 4):
... def f():
... return i*i
... fs.append(f())
... return fs
...
>>>f1, f2, f3 = count()
>>>[f1, f2, f3]
[1, 4, 9]
在闭包中,使用f
时,函数内的函数体并未执行,而是在count()
运行结束后,再次回到f
中实现该函数;
而f()
则表示,在count()
函数内执行f()
函数。
类似的,如果一定要引用循环变量怎么办?方法是再创建一个函数,用该函数的参数绑定循环变量当前的值,无论该循环变量后续如何更改,已绑定到函数参数的值不变:
>>> def count():
... fs = []
... def f(i):
... return lambda :i*i
... for n in range(1, 4):
... fs.append(f(n))
... return fs
...
>>> f1, f2, f3 = count()
>>> f1()
1
>>> f2()
4
>>> f3()
9