15. 使用生成器函数实现可迭代对象

要求：实现一个可迭代对象的类，它能迭代出给定范围内的所有素数（只能被1和自己整除的数，2除外）：

pn = PrimeNumbers(1, 30)

for k in pn:
    print(k)

输出结果：

2 3 5 7 11 13 17 19 23 29

解决方案：

将该类的__iter__()方法实现为生成器函数，每次yield返回一个素数。

---

• 对于生成器函数：

在Python中，使用了yield关键字的函数被称为生成器generator。

跟普通函数不同的是，生成器是一个返回迭代器的函数，只能用于迭代操作，简单点理解生成器就是一个迭代器。

在调用生成器运行的过程中，每次遇到yield时函数会暂停并保存当前所有的运行信息，返回yield的值，并在下一次执行next()方法时从当前位置继续运行。

调用一个生成器函数，返回的是一个迭代器对象。

生成器分类：

1、生成器函数：常规函数定义，但是使用yield语句而不是return语句返回结果，yield语句一次返回一个结果，可以使用多次，在每个结果中间，挂起函数的状态，以便下次从它离开的地方继续执行。

2、生成器表达式：类似列表推导式，生成器返回按需产生结果的一个对象，而不是一次构建一个完整的结果列表。

生成器优点:

在需要的时候才产生结果，不是立即产生结果。

---

解决方案：

from collections.abc import Iterable

class PrimeNumbers(Iterable):
    
    def __init__(self, a, b):
        self.a = a
        self.b = b

    def __iter__(self):
        for k in range(self.a, self.b + 1):
            if self.is_prime(k):
                yield k

    def is_prime(self, k):
        
        return False if k < 2 else all(map(lambda x: k % x, range(2, k)))

pn = PrimeNumbers(1, 50)

for n in pn:
    print(n, end= ' ')
    
2 3 5 7 11 13 17 19 23 29 31 37 41 43 47                #结果

生成器对象可以自动维护迭代状态，每次yield才返回一个素数。

---