如何把一个类作为一个迭代器对象使用

最新推荐文章于 2022-05-03 23:11:25 发布

转载最新推荐文章于 2022-05-03 23:11:25 发布 · 867 阅读

文章标签：

#迭代器

python 专栏收录该内容

5 篇文章

订阅专栏

本文深入讲解了如何在Python中创建自定义迭代器，包括必要的iter()和next()方法的实现，以及如何通过抛出StopIteration异常来控制迭代次数，避免无限循环。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.把一个类作为一个迭代器使用需要在类中实现两个方法 iter() 与 next()
iter_() 方法返回一个特殊的迭代器对象
next() 方法会返回下一个迭代器对象


    class MyNumbers:
      def __iter__(self):
        self.a = 1
        return self
     
      def __next__(self):
        x = self.a
        self.a += 1
        return x
     
    myclass = MyNumbers()
    myiter = iter(myclass)
     
    print(next(myiter))
    print(next(myiter))
    print(next(myiter))
    print(next(myiter))
    print(next(myiter))
    ```
    执行输出结果为：
    
    1
    2
    3
    4
    5
    **2.StopIteration**
    StopIteration 异常用于标识迭代的完成，防止出现无限循环的情况，在 __next__() 方法中我们可以设置在完成指定循环次数后触发 StopIteration 异常来结束迭代。
    
    在 20 次迭代后停止执行：

    实例(Python 3.0+)
        ```
    class MyNumbers:
      def __iter__(self):
        self.a = 1
        return self
     
      def __next__(self):
        if self.a <= 20:
          x = self.a
          self.a += 1
          return x
        else:
          raise StopIteration
     
    myclass = MyNumbers()
    myiter = iter(myclass)
     
    for x in myiter:
      print(x)

执行输出结果为：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20