python3 函數迭代器

迭代器協議:

迭代器協議,是指對象(實例)能夠使用next函數獲取下一項數據,在沒有下一項數據之前觸發一個StopIteration異常來終止迭代


next(it)  對應__next__(self)方法

iter(obj) 對應__iter__(self)方法,通常返回一個可迭代對象



class odd:

    def __init__(self,begin,end):

        self.beg = begin

        self.end = end 

        self.cur = begin   #數據的當前位置

    

    def __next__(self):

        """print("next被調用")"""

        if self.cur >= self.end:

            raise StopIteration

        #此判斷獲取一個奇數,

        if self.cur % 2 == 0:

            self.cur += 1


        r = self.cur

        self.cur += 1   #步長

        return r

    

    def __iter__(self):

        """__iter__被調用,返回自己作爲迭代器,每次返回一個可迭代對象,調用一次__iter__"""    

        self.cur = self.beg

        return self


o = odd(5,10)

for x in o:

    print(x)


print([x for x in o])


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章