写一个斐波那契数列的迭代器

Python中没有“迭代器”这个类，因此具有以下两个特性的类都可以称为“迭代器”类：

1、有__next__()方法，返回容器的下一个元素或抛出StopIteration异常

2、有__iter__()方法，返回迭代器本身

斐波那契数列迭代器代码：

class Fib:
    """迭代器"""

    def __init__(self, num):
        self.num = num
        self.a, self.b = 0, 1
        self.idx = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.idx < self.num:
            self.a, self.b = self.b, self.a + self.b
            self.idx += 1
            return self.a
        raise StopIteration()


def main():
    """主函数"""
    for val in Fib(20):
        print(val)



if __name__ == '__main__':
    main()