python利用定義類實現迭代器協議
阿新 • • 發佈:2018-11-05
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