# -*- 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()函数直接得到当前计数的值。
迭代器在这里就充当计数器的作用。因为每计数一次,迭代器的指针就往后移动一个单位。