如何使用生成器函数实现可迭代对象

问题举例：

实现一个可迭代对象的类，它能迭代出给定范围内 的所有素数：

pn = PrimeNumbers（1， 30）

for x in pn:

　　print(x)

 

输出结果：

2 3 5 7 11 13 17 19 23 29

 

解决思路

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

 

代码

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, 30)
for n in pn:
    print(n)

 

参考资料：python3实用编程技巧进阶