python中实现迭代器（iterator）的方法示例

下面是 "Python中实现迭代器（iterator）的方法示例" 的完整攻略：

什么是迭代器（iterator）

在Python中，迭代器（iterator）是一种确保我们可以按照特定顺序访问集合中每个元素的对象。它是用来迭代或遍历可迭代对象的工具，如列表、元组、字典等。迭代器对象可以使用 next() 方法来逐个返回集合中的数据项，当所有元素都已访问完毕时，会抛出 StopIteration 异常。

如何实现迭代器（iterator）

Python中实现迭代器的方法很简单，只需要在类中实现 __iter__() 和 __next__() 方法即可。

__iter__() 方法

在 Python 中，任何具有 __iter__() 方法的对象都可以用于 迭代。__iter__() 方法是一种用于返回迭代器（iterator）自身的方法，在该方法中我们以任何方式定义迭代器的初始化状态。

__iter__() 方法应该返回尽可能多的“新”迭代器对象，以便可以同时迭代同一集合的多个部分。

下面是使用 __iter__() 方法创建迭代器的示例：

class MyIterator:
    def __iter__(self):
        self.num = 0
        return self

    def __next__(self):
        if self.num < 10:
            self.num += 1
            return self.num
        else:
            raise StopIteration

上面的示例中，我们创建了一个名为 MyIterator 的迭代器类。在 __iter__() 方法中，我们将 num 变量初始化为 0，并将自身作为迭代器对象返回。在 __next__() 方法中，我们使用 num 变量访问集合中的每个元素，并使用 raise StopIteration 语句来通知迭代器已到达集合的末尾。

__next__() 方法

__next__() 方法是用于从迭代器中返回下一项的方法。在该方法中，我们定义了如何返回一个新值，并在达到一定条件时通知迭代器结束迭代。

下面是另一种使用 __next__() 方法实现迭代器的示例：

class Colors:
    def __init__(self):
        self.colors = ['red', 'orange', 'yellow', 'green', 'blue', 'purple']
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        if self.index == len(self.colors):
            raise StopIteration
        color = self.colors[self.index]
        self.index += 1
        return color

上面的示例中，我们创建了一个名为 Colors 的迭代器类。在 __iter__() 方法中，我们返回自身，而在 __next__() 方法中，我们从 colors 列表中返回每个颜色的值，同时通过 raise StopIteration 语句来通知迭代器结束迭代（当 index 值达到 len(colors) 时，就说明已经迭代完了所有元素）。

总结

通过本文的介绍，我们了解了Python中迭代器的基本原理和使用方法。迭代器是一种用于迭代或遍历数据结构元素的工具，我们只需要实现 __iter__() 和 __next__() 两种方法即可以创建一个迭代器。下次你需要遍历一个可迭代的集合时，不妨使用自己实现的迭代器来完成。