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