Python iter：解锁自定义可迭代对象的奥秘

01-24 2437阅读

Python iter：解锁自定义可迭代对象的奥秘

在Python编程的世界里，迭代是一项极为常见且重要的操作。我们经常使用for循环遍历列表、元组等内置可迭代对象，但你是否想过如何创建自定义的可迭代对象呢？这就不得不提到Python中的__iter__方法。

可迭代对象与迭代器的概念

在深入探讨__iter__方法之前，先了解一下可迭代对象和迭代器的概念。可迭代对象是指实现了__iter__方法的对象，而迭代器是实现了__iter__和__next__方法的对象。__iter__方法返回一个迭代器，而__next__方法返回迭代的下一个值，当没有更多元素时，会抛出StopIteration异常。

自定义可迭代对象

下面通过一个简单的例子来展示如何使用__iter__方法创建自定义可迭代对象。假设我们要创建一个可迭代对象，用于生成指定范围内的偶数。

Python iter：解锁自定义可迭代对象的奥秘

class EvenNumbers:
    def __init__(self, start, end):
        self.start = start if start % 2 == 0 else start + 1
        self.end = end

    def __iter__(self):
        return self.EvenIterator(self.start, self.end)

    class EvenIterator:
        def __init__(self, start, end):
            self.current = start
            self.end = end

        def __iter__(self):
            return self

        def __next__(self):
            if self.current <= self.end:
                result = self.current
                self.current += 2
                return result
            else:
                raise StopIteration

# 使用自定义可迭代对象
even_nums = EvenNumbers(1, 10)
for num in even_nums:
    print(num)

在这个例子中，EvenNumbers类是一个可迭代对象，它的__iter__方法返回了一个EvenIterator类的实例，而EvenIterator类是一个迭代器，它实现了__next__方法，用于生成下一个偶数。