要求:实现一个可迭代对象的类,它能迭代出给定范围内的所有素数(只能被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
才返回一个素数。