Python学习笔记(1)，闭包和变量作用域

def lazy_sum(*args):
    def sum():
        ax = 0
        for n in args:
            ax = ax + n
        return ax
    return sum

f = lazy_sum(1, 2, 3, 4)
print(f()) # output: 10

'''
我们在函数lazy_sum中又定义了函数sum，并且，内部函数sum可以引用外部函数lazy_sum的参数和局部变
量，当lazy_sum返回函数sum时，相关参数和变量都保存在返回的函数中，这种称为“闭包（Closure）”的
程序结构拥有极大的威力。 
'''


'''
闭包注意事项
'''
def count():
    fs = []
    for i in range(1, 4):
        def f():
            return i*i
        fs.append(f)
    return fs

f1, f2, f3 = count()
print(f1()) # 9
print(f2()) # 9
print(f3()) # 9
# 全部都是9！原因就在于返回的函数引用了变量i，但它并非立刻执行。等到3个函数都返回时，
# 它们所引用的变量i已经变成了3，因此最终结果为9。
'''返回闭包时牢记一点：返回函数不要引用任何循环变量，或者后续会发生变化的变量。
'''
def count2():
    def f(j):
        def g():
            return j*j
        return g
    fs = []
    for i in range(1, 4):
        fs.append(f(i))  # f(i)立刻被执行，因此i的当前值被传入f()
    return fs
ff1, ff2, ff3 = count2()
print(ff1())
print(ff2())
print(ff3())


'''利用闭包返回一个计数器函数，每次调用它返回递增整数：'''
def createCounter():
    i = 0
    def counter():
        nonlocal i
        i += 1
        return i
    return counter
# 测试:
counterA = createCounter()
print(counterA(), counterA(), counterA(), counterA(), counterA())  # 1 2 3 4 5
counterB = createCounter()
if [counterB(), counterB(), counterB(), counterB()] == [1, 2, 3, 4]:
    print('测试通过!')
else:
    print('测试失败!')


def createCounter2():
    x = [0]
    def counter2():
        x[0] += 1
        return x[0]
    return counter2

counterA2 = createCounter2()
print(counterA2(), counterA2(), counterA2(), counterA2(), counterA2())  # 1 2 3 4 5
counterB2 = createCounter2()
if [counterB2(), counterB2(), counterB2(), counterB2()] == [1, 2, 3, 4]:
    print('测试通过!')
else:
    print('测试失败!')

'''
在counter()函数内的 i 是局部变量，counter()不能对外部变量进行修改，如果不声明nonlocal，
会报错为局部变量未定义。如果将 i 声明为 nonlocal，会往上一层作用域中寻找 i。
也可以吧外部变量申明为 list，内部函数可以更新list里的内容，但是没有改变 list.
'''