6、Python 迭代器与生成器全解析

Python 迭代器与生成器全解析

1. 迭代器基础

在 Python 中，迭代器是实现了 __iter__() 和 __next__() 方法的对象。 __iter__() 方法返回迭代器对象本身，而 __next__() 方法返回序列中的下一个元素。当没有更多元素时， __next__() 方法会引发 StopIteration 异常。

1.1 简单迭代器示例

以下是一个简单的类，展示了如何实现迭代器：

class Node:
    def __init__(self, value):
        self._value = value
        self._children = []

    def __repr__(self):
        return 'Node({!r})'.format(self._value)

    def add_child(self, node):
        self._children.append(node)

    def __iter__(self):
        return iter(self._children)

# 示例
if __name__ == '__main__':
    root = Node(0)
    child1 = Node(1)
    child2 = Node(2)
    root.add_child