实际案例
实现一个可迭代对象的类,它能迭代出给定范围内所有素数:
- 1
- 2
- 3
输出结果为:
- 1
解决方案:将该类的__iter__方法实现成生成器函数,每次yield返回一个素数。
有了解决方案后,我们首先要了解一下什么是生成器函数。让我们先看看如下代码:
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
其输出结果为:
- 1
诶,我们调用了f()却没有打印输出我们设想的结果,这是为什么呢?我们现在来看看这个输出结果,它表示调用f()后的返回结果类型为generator类型,即返回的结果为生成器对象。那么,我们将任何包含yield 语句的函数称为生成器(函数)。当然,生成器不仅仅是包含yield语句的函数,更多关于生成器的知识请参阅廖雪峰的Python教程。
yield语句不会像return那样返回值,而是每次产生多个值。每次产生一个值(使用yield语句),函数就会被冻结:即函数停在那点等待被重新唤醒。函数被重新唤醒后就从停止的那点开始执行。
将yield语句的知识补充完后,我们再回到生成器这个知识点。上面我们讲到我们调用f()后返回一个生成器对象,其和迭代器对象类似,它也可以实现next()。(注:Python 3.X版本中为__next__()。)
那我们就在代码中实现next():
- 1
- 2
- 3
其结果为:
- 1
- 2
- 3
- 4
- 5
- 6
通过上述代码,我们又可发现g(生成器对象)也是一个迭代器对象。因此,我们就可以利用这一点解决我们的实际案例,代码如下:
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
输出结果为: