如何使用生成器函數(shù)實現(xiàn)可迭代對象

問題
實現(xiàn)一個可迭代對象的類,它能迭代出給定范圍內(nèi)所有的素數(shù)

解決方案:

  • 將該類的iter方法實現(xiàn)成生成器函數(shù),每次yield返回一個素數(shù)
def f():
    print 'in f().1'
    yield 1

    print 'in f().2'
    yield 2

    print 'in f().3'
    yield 3

此處直接調(diào)用f()并不會得到值

g = f()

生成器函數(shù)與迭代器對象同樣擁有next()

print f().next()

雖然返回了yield,但是保存了程序運行的狀態(tài)

生成器對象也是一個可迭代對象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
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容