1. 程式人生 > >python利用定義類實現迭代器協議

python利用定義類實現迭代器協議

1.具有__ next__和__ iter__方法的物件即被視為迭代器,其中next方法每執行一次返回迭代的下一項,而iter方法可以把物件變成迭代器.
執行for迴圈的本質即是先對物件使用iter方法, 然後不斷使用next方法,直到報出StopIteration. 故可直接在類中定義這兩個方法,從而實現迭代器協議.
下例:

class Test:
    def __init__(self):
        self.a = 1
        self.b = 1
        self.c = 0

    def __iter__(self):
        return
self def __next__(self): if self.b > 20: raise StopIteration if self.c <= 1: self.c += 1 return 1 self.a, self.b = self.b, self.a + self.b return self.b t1 = Test() for i in t1: print(i) #result 1 1 2 3 5 8 13 21