Python迭代器和生成器

最新推荐文章于 2024-10-08 12:37:10 发布

代码输入中...

最新推荐文章于 2024-10-08 12:37:10 发布

阅读量428

点赞数 1

文章标签： python 深度学习开发语言人工智能测试工具

本文链接：https://blog.csdn.net/m0_59485658/article/details/128164407

版权

本文详细介绍了Python中的迭代器和生成器。迭代器需要实现__iter__()和next()方法，可自定义迭代器。可迭代对象与迭代器对象有时是独立的，但有些可迭代对象本身就是迭代器。生成器是特殊的迭代器，通过生成器函数创建，使用yield表达式逐次返回结果。生成器表达式提供惰性计算，节省内存。递归生成器允许生成器递归调用。此外，还讨论了生成器的send()和close()方法。

摘要由CSDN通过智能技术生成

在Python中，很多对象都是可以通过for语句来直接遍历的，例如list、string、dict等等，这些对象都可以被称为可迭代对象。至于说哪些对象是可以被迭代访问的，就要了解一下迭代器相关的知识了。

迭代器

迭代器对象要求支持迭代器协议的对象，在Python中，支持迭代器协议就是实现对象的__iter__()和next()方法。其中__iter__()方法返回迭代器对象本身；next()方法返回容器的下一个元素，在结尾时引发StopIteration异常。

iter()和next()方法

这两个方法是迭代器最基本的方法，一个用来获得迭代器对象，一个用来获取容器中的下一个元素。

对于可迭代对象，可以使用内建函数iter()来获取它的迭代器对象：

例子中，通过iter()方法获得了list的迭代器对象，然后就可以通过next()方法来访问list中的元素了。当容器中没有可访问的元素后，next()方法将会抛出一个StopIteration异常终止迭代器。

其实，当我们使用for语句的时候，for语句就会自动的通过__iter__()方法来获得迭代器对象，并且通过next()方法来获取下一个元素。

自定义迭代器

了解了迭代器协议之后，就可以自定义迭代器了。

下面例子中实现了一个MyRange的类型，这个类型中实现了__iter__()方法，通过这个方法返回对象本身作为迭代器对象；同时，实现了next()方法用来获取容器中的下一个元素，当没有可访问元素后，就抛出StopIteration异常。

class MyRange(object):
    def __init__(self, n):
        self.idx = 0
        self.n = n
        
    def __iter__(self):
        return self
        
    def next(self):
        if self.idx < self.n:
            val = self.idx
            self.idx += 1
            return val
        else:
            raise StopIteration()

这个自定义类型跟内建函数xrange很类似，看一下运行结果：