Python3 - 使用生成器创建新的迭代模式

问题

实现一个自定义迭代模式，与普通的内置函数比如 range(), reversed() 不一样。

解决方案

使用生成器函数，定义一个实现某个范围内浮点数的生成器，比如：

def frange(start, stop, increment):
    x = start
    while x < stop:
        yield x
        x += increment

可以用for循环迭代这个函数或者使用其他可接受可迭代对象的函数(如 sum() , list() 等)。示例：

for n in frange(0, 3, 0.5):
    print(n, end='  ')
0  0.5  1.0  1.5  2.0  2.5 

float_list = list(frange(0, 1, 0.125))
print(float_list)
[0, 0.125, 0.25, 0.375, 0.5, 0.625, 0.75, 0.875]

讨论

定义的函数中，通过 yield 语句即可将其转换为一个生成器。 跟普通函数不同的是，生成器只能用于迭代操作。示例演示这样的迭代函数，底层的工作机制：

def countdown(n):
    print('Starting to count from', n)
    while n > 0:
        yield n
        n -= 1
    print('Done !')

c = countdown(2)
print(c)
<generator object countdown at 0x10cfa8b48>

print(next(c))
Starting to count from 2
2

print(next(c))
1

print(next(c))
Done!
Traceback (most recent call last):
    File "<stdin>", line 1, in <module>
StopIteration

一个生成器函数主要特征是它只会在迭代中使用到 next 操作。 一旦生成器函数返回退出，迭代终止。在迭代中通常使用的for语句，它会自动处理这些细节，所以无需担心。