4.4实现迭代器协议

问题:

​ 你想构建一个能支持迭代操作的额自定义对象,并希望找到一个能实现迭代协议的简单方法。

解决方案:

​ 目前为止,在对象上实现迭代最简单的方式是使用一个生成器函数。可以实现一个深度优先遍历树形节点的生成器。下面是实现代码

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)

    def depth_first(self):
        yield self
        for c in self:
            yield from c.depth_first()


if __name__ == '__main__':
    root = Node(0)
    child1 = Node(1)
    child2 = Node(2)
    root.add_child(child1)
    root.add_child(child2)
    child1.add_child(Node(3))
    child1.add_child(Node(4))
    child1.add_child(Node(5))
    for ch in root.depth_first():
        print(ch)
        '''
        Node(0)
        Node(1)
        Node(3)
        Node(4)
        Node(5)
        Node(2)
        '''

​ 在这段代码中,depth_first()方法简单直观,它首先返回自己本身并迭代每一个子节点并通过子节点的depth_first()方法(使用yeild from语句)返回对应元素。

posted @ 2022-03-17 14:15  qiupeng  阅读(33)  评论(0)    收藏  举报