Pyhton函数返回函数

# -*- coding: utf-8 -*-

def createCounter():
    def add_num():
        n = 0
        while True:
            n = n + 1
            yield n
    y = add_num()
        
    def counter():
        return next(y)
    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('测试失败!')

Python原函数返回的函数在原函数执行时并未执行，而是在有需要时执行。

此外，返回的函数不要引用原函数的任何可能变化的量（如循环控制变量）

在上述Conter例子中

一开始的想法是：

def createCounter():
    a = 1     
    def counter():
        i = a
        a = a + 1
        return i
    return counter

但是测试过程中发现可执行i=a，但不能执行a=a+1。即可访问原函数变量，但是不能改变原函数的变量（此时会提示a未定义）。

因此后面想法便是在原函数执行时创建一个从0开始无限整数序列（无限迭代器），这样每次就可通过next()函数直接得到当前计数的值。

迭代器在这里就充当计数器的作用。因为每计数一次，迭代器的指针就往后移动一个单位。