Python进阶之:iter

被刻印的时光 ゝ 提交于 2019-12-20 13:35:24
'''
在某些情况下,我们希望实例对象可被用于 for...in 循环,这时我们需要在类中定义 __iter__ 和 next(在 Python3 中是 __next__)方法,
其中,__iter__ 返回一个迭代对象,next 返回容器的下一个元素,在没有后续元素时抛出 StopIteration 异常。
'''

class Fib(object):
    def __init__(self):
        self.a,self.b = 0,1

    def __iter__(self): # 返回迭代器本身
        return self

    def __next__(self): # 返回容器的下一个元素
        self.a,self.b = self.b,self.a + self.b
        return self.a

    def __getitem__(self, item):
        self.item = item


fib = Fib()
for i in fib:
    if i > 1000:
        break
    print(i)

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!