问题:
你想实现一个自定义迭代模式,跟普通的内置函数比如range(),reverse的()不一样。
解决方案:
如果你想实现一种新的迭代模式,使用一个生成函数来定义它。下面是 一个生产某个范围内浮点数的生成器。
def frange(start, stop, increment):
x = start
while x < stop:
yield x
x += increment
for n in frange(1, 10, 3):
print(n)
print(list(frange(1, 10, 3)))#->[1, 4, 7]
讨论:
一个函数中需要有一个yeild语句即可将其转换为一个生成器。跟普通函数不同的是,生成器只能用于迭代操作。下面是一个实验,向你展示这样的函数底层工作机制。
def countdown(n):
print('Staring to count from', n)
while n > 0:
yield n
n-=1
print('Done!')
c=countdown(3)
print(c)
print(next(c))# ->3
print(next(c))# ->2
print(next(c))# ->1
print(next(c))
'''
line 25, in <module>
StopIteration
'''
一个生成器函数主要特征是它只会回应在迭代中使用到的next操作,一旦生成器函数返回退出,迭代终止,我们在迭代中通常使用的for语句会自动处理这些细节,所以你无需担心。