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

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

解决方案:

  • 将该类的iter方法实现成生成器函数,每次yield返回一个素数
def f():
    print 'in f().1'
    yield 1

    print 'in f().2'
    yield 2

    print 'in f().3'
    yield 3

此处直接调用f()并不会得到值

g = f()

生成器函数与迭代器对象同样拥有next()

print f().next()

虽然返回了yield,但是保存了程序运行的状态

生成器对象也是一个可迭代对象iter返回它本身。
也是一个迭代器对象,拥有next()方法

class PrimeNumber:
    def __init__(self,start,end):
        self.start = start
        self.end = end
    def isPrimeNumber(self,k):
            if k < 2:
                return False
            for i in xrange(2, k):
                if k % i == 0:
                    return False
            return True

    def __iter__(self):
        for k in xrange(self.start, self.end +1):
            if self.isPrimeNumber(k):
                yield k

for x in PrimeNumber(1,30):
    print x
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容