首页 > 代码库 > Python 函数式编程(3) —— 闭包

Python 函数式编程(3) —— 闭包

内层函数引用了外层函数的变量(参数也算变量),然后返回内层函数的情况,称为闭包(Closure)。闭包的特点是返回的函数还引用了外层函数的局部变量,所以,要正确使用闭包,就要确保引用的局部变量在函数返回后不能变。

举例:希望一次返回3个函数,分别计算1x1,2x2,3x3:
def count():
    fs = []
    for i in range(1, 4):
        def f():
            return i*i
        fs.append(f)
    print(fs)
    return fs

f1, f2, f3 = count()
print(f1(), f2(), f3())
# 结果全部都是 9。原因就是当count()函数返回了3个函数时,这3个函数所引用的变量 i 的值已经变成了3。
# 由于f1、f2、f3并没有被调用,所以,此时他们并未计算 i*i,当 f1 被调用时:9。
# 因此,返回函数不要引用任何循环变量,或者后续会发生变化的变量。

返回闭包不能引用循环变量,请改写count()函数,让它正确返回能计算1x1、2x2、3x3的函数。
def count1():
    fs = []
    for i in range(1, 4):
        def f(j):
            def g():
                return j*j
            return g
        result = f(i)
        fs.append(result)
    print(fs)
    return fs
a1, a2, a3 = count1()
print(a1(), a2(), a3())

 

Python 函数式编程(3) —— 闭包