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