python迭代器相关方法

红糖果仁沙琪玛

已于 2022-06-21 20:36:03 修改

阅读量491

点赞数

CC 4.0 BY-SA版权

分类专栏： Python学习日志文章标签： python 开发语言

于 2022-06-21 16:51:01 首次发布

本文链接：https://blog.csdn.net/K_ZhJ18/article/details/125394104

Python学习日志专栏收录该内容

6 篇文章

订阅专栏

本文详细介绍了Python中的迭代器和生成器。通过`iter()`函数获取迭代器，`next()`方法执行迭代，演示了字典、列表等可迭代对象的迭代过程。同时，讲解了迭代器生成式和`yield`关键字的应用，展示了如何创建自定义迭代器，并利用`StopIteration`异常控制迭代结束。文章还提供了实例代码帮助理解这些概念。

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

iter、next

首先最主要的有两个方法，生成迭代器iter()，执行迭代next()。iter生成迭代器会记下每次执行后的位置，而next会根据此位置继续执行

凡是可迭代对象，如字典、列表、元组、字符串···都可以使用iter来生成迭代器，其返回类型为iteration。其中，若对字典进行iter，则会将其键生成迭代器，如下：

In [16]: dic={'a':1,'b':2,'c':3}
In [17]: A=iter(dic)
In [18]: type(A)
Out[18]: dict_keyiterator

In [19]: next(A)
Out[19]: 'a'

In [21]: next(A)
Out[21]: 'b'

In [22]: next(A)
Out[22]: 'c'

也可以用循环体对迭代对象进行输出

In [24]: dic={'a':1,'b':2,'c':3}

In [25]: B=iter(dic)

In [26]: for i in B:
    ...:     print(i)
    ...:
a
b
c

迭代器生成式

In [27]: C=(i for i in [1,2,3,4,5,6])

In [28]: type(C)
Out[28]: generator

In [29]: next(C)
Out[29]: 1

In [30]: next(C)
Out[30]: 2

类似列表生成式，只需将[]换为()即可

魔术方法及stopinteration

为了防止迭代对象一直进行迭代，可以使用stopinteration来终止迭代

class Interation():
    def __iter__(self):
        self.a=0
        return self
    def __next__(self):
        if self.a<5:
            self.a+=1
            return self.a
        else:
            raise StopIteration

A=Interation()
print(A)
B=iter(A)
print(B)
while True:
    print(next(B))

<__main__.Interation object at 0x0000020953D2C910>
<__main__.Interation object at 0x0000020953D2C910>
1
2
3
4
5

yeild

在函数中，将return修改为yeild，也可生成一个迭代对象