迭代器和生成器

  • 迭代器是实现了迭代器协议的对象。

    • Python中没有像protocol或interface这样的定义协议的关键字。
    • Python中用魔术方法表示协议。
    • __iter__和__next__魔术方法就是迭代器协议。
    1. class Fib(object):
    2.     """迭代器"""
    4.     def __init__(self, num):
    5.         self.num = num
    6.         self.a, self.b = 0, 1
    7.         self.idx = 0
    9.     def __iter__(self):
    10.         return self
    12.     def __next__(self):
    13.         if self.idx < self.num:
    14.             self.a, self.b = self.b, self.a + self.b
    15.             self.idx += 1
    16.             return self.a
    17.         raise StopIteration()

  • 生成器是语法简化版的迭代器。

    1. def fib(num):
    2.     """生成器"""
    3.     a, b = 0, 1
    4.     for _ in range(num):
    5.         a, b = b, a + b
    6.         yield a

  • 生成器进化为协程。

    生成器对象可以使用send()方法发送数据，发送的数据会成为生成器函数中通过yield表达式获得的值。这样，生成器就可以作为协程使用，协程简单的说就是可以相互协作的子程序。

    ```Python def calc_avg():

    1. """流式计算平均值"""
    2. total, counter = 0, 0
    3. avg_value = None
    4. while True:
    5.     value = yield avg_value
    6.     total, counter = total + value, counter + 1
    7.     avg_value = total / counter

gen = calc_avg() next(gen) print(gen.send(10)) print(gen.send(20)) print(gen.send(30)) ```