【python】迭代器与生成器

星眺北海

于 2024-08-12 15:20:00 发布

阅读量161

点赞数 7

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/longforone/article/details/141132628

版权

python 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

Python的迭代器和生成器

在Python中，迭代器和生成器是两种非常强大的工具，它们允许我们以一种高效且优雅的方式处理数据集合。本文将探讨迭代器和生成器的概念、它们之间的联系以及如何在实际编程中使用它们.

迭代器（Iterator）

迭代器是一个可以记住遍历位置的对象。迭代器对象必须实现两个方法：__iter__() 和 __next__()。

__iter__() 方法返回迭代器对象本身。
__next__() 方法返回迭代器的下一个元素。

Python中的迭代器协议要求实现这两个方法。

其实就是 iter 和next调用的魔术方法。

示例代码

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index < len(self.data):
            result = self.data[self.index]
            self.index += 1
            return result
        else:
            raise StopIteration

通过实现迭代器协议，Python提供了一种统一的方法来遍历各种数据类型。这意味着无论数据存储在列表、元组、字典、集合还是任何自定义的数据结构中，只要它们遵循迭代器协议，我们就可以以相同的方式处理它们。

例如现在我们可以直接对自定义的数据结构使用for进行遍历了

生成器（Generator）

生成器是一种特殊的迭代器，它使用yield语句来产生值。每次调用生成器的__next__()方法时，生成器会从上次yield的位置继续执行。

使用send()方法

在使用生成器时，既可以用像迭代器一样使用next()方法获取值，也可以使用send()方法。

与next()方法相比，send()方法允许我们向生成器发送值。而这正是生成器的含义，我们像其中传递参数，然后生成器根据参数区生成一个结果响应给我们，而不是像迭代器一样直接从数据集合中获取值。

示例代码

下面是一个计算平方的生成器示例

def my_generator():
    val = yield
    while True:
        val = yield val * val


g = my_generator()
print(next(g))
print(g.send(2))
print(g.send(10))

# 输出如下
# None
# 4
# 100