可迭代对象(iter()和next())

最新推荐文章于 2025-01-13 18:20:24 发布

原创最新推荐文章于 2025-01-13 18:20:24 发布 · 139 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #java #算法 #leetcode #c++

本文介绍如何通过实现__iter__()和__next__()方法使自定义类支持for...in循环。通过一个斐波那契数列生成器的具体示例，展示了如何控制循环终止条件，并通过抛出StopIteration异常来结束循环。

如果一个类想被用于for ... in循环，类似list或tuple那样，就必须实现一个__iter__()和__next__()方法，该方法返回一个迭代对象
然后，Python的for循环就会不断调用该迭代对象的__next__()方法拿到循环的下一个值，直到遇到StopIteration错误时退出循环

    class Fib(object):
        def __init__(self):
            self.a, self.b = 0, 1 # 初始化两个计数器a，b

        def __iter__(self):
            return self # 实例本身就是迭代对象，故返回自己

        def __next__(self):
            self.a, self.b = self.b, self.a + self.b # 计算下一个值
            if self.a > 100000: # 退出循环的条件
                raise StopIteration()
            return self.a # 返回下一个值
            
    for n in Fib():
        print(n)
    
　　#输出：1
    #输出：1
    #输出：2
    #输出：3
    #输出：5
    #输出：...
    #输出：46368
    #输出：75025