如何使用生成器函数实现可迭代对象
问题举例:
实现一个可迭代对象的类,它能迭代出给定范围内 的所有素数:
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实用编程技巧进阶