如何在 Python 中创建和使用迭代器？

在 Python 中创建迭代器

在 Python 中构建迭代器需要实现迭代器协议，该协议指定了两个基本方法： __iter__() 和__下一个__（）。这些方法定义了对象如何初始化和迭代一系列值。

理解迭代器协议

在 __iter__() 中，返回迭代器对象，即通常在循环开始时隐式调用。 __next__() 是返回序列中下一个值的主要方法。对于 Python 2 用户，此方法称为 next()。当所有值都用完时， __next__() 会引发 StopIteration 异常，该异常会循环构造 capture 来终止迭代。

示例：计数器迭代器

让我们创建一个简单的生成指定范围内的值的计数器迭代器：

class Counter:
    def __init__(self, low, high):
        self.current = low - 1
        self.high = high

    def __iter__(self):
        return self

    def __next__(self): # Python 2: def next(self)
        self.current += 1
        if self.current < self.high:
            return self.current
        raise StopIteration

# Usage:
for c in Counter(3, 9):
    print(c)

这将产生以下结果输出：

3
4
5
6
7
8

将生成器用于迭代器

生成器提供了另一种创建迭代器的机制。生成器函数一次产生一个值，有效地实现了迭代器协议。

def counter(low, high):
    current = low
    while current < high:
        yield current
        current += 1

# Usage:
for c in counter(3, 9):
    print(c)

上面的代码产生与 Counter 类相同的输出。

其他资源

要全面了解迭代器，请参阅 David Mertz 的文章“迭代器和简单”等资源发电机。”

以上是如何在 Python 中创建和使用迭代器？的详细内容。更多信息请关注PHP中文网其他相关文章！