码迷,mamicode.com
首页 > 其他好文 > 详细

3-3利用生成器实现可迭代对象

时间:2018-04-14 11:18:55      阅读:107      评论:0      收藏:0      [点我收藏+]

标签:函数   als   AC   img   内容   start   justify   范围   com   

技术分享图片

包含yield语句的函数就是生成器函数。数里有yield关键字,则是生成器,生成器内置有__iter__方法,只不过调用__iter__返回的是生成器本身,利用这一特性,可以创建一个可迭代对象的类。

生成器即实现了,next迭代器对象,同时也实现了__iter__()可迭代对象,返回的是他自身

>>> def f():
    print(1111)
    yield 1          #当调用next()时 yield会把后面的内容返回,但保存当前状态,下次调用next()时继续执行。
    print(2222)
    yield 2
    print(3333)
    yield 3
>>> g = f()
>>> g.next()
1111
1
>>> g.next()
2222
2
>>> g.next()
3333
3
>>> g.next()

Traceback (most recent call last):
  File "<pyshell#57>", line 1, in <module>
    g.next()
StopIteration

用迭代方式:

>>> g = f()
>>> for x in g:
    print x

    
1111
1
2222
2
3333
3
>>> 

和迭代器实现了相同的功能。

>>> g.__iter__() is g
True

验证了生成器实现了__iter__()可迭代对象,返回的是他自身

将可迭代对象的__iter__()方法实现在生成器函数。当__iter__()运行时不会直接返回,而是会返回一个包含next()方法的生成器对象。每次通过yield返回

class PrimeNum():   #如果是定义成 class PrimeNum:  也能正确执行。
    def __init__(self,start,end):
        self.start = start
        self.end = end

    def isPrime(self,k):   #类中定义判断是否是素数的方法
        if k <= 2:
            return False
        for x in xrange(2,k):
            if k % x == 0:
                return False
        return True

    def __iter__(self):
        for x in xrange(self.start,self.end +1): #判断输入的范围如果是素数,用yield返回当前值,等待下迭代,继续返回
            if self.isPrime(x):
                yield x

primenum = PrimeNum(0,100)

for x in primenum:
    print (x),

输出:3 5 7 11 13 17 19 23 29 31 37 41 43 47 53 59 61 67 71 73 79 83 89 97

3-3利用生成器实现可迭代对象

标签:函数   als   AC   img   内容   start   justify   范围   com   

原文地址:https://www.cnblogs.com/smulngy/p/8830448.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!