委托给list对象 --委托迭代.

问题:构建了一个自定义的容器对象,其内部持有列表、元组或其他可迭代对象。我们想让自己的新容器能够完成迭代操作。

解决方案:一般来说,我们所要做的就是定义一个__iter__()方法,将迭代请求委托到对象内部持有的容器上

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

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

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

    def __iter__(self):
        """ 将类实例中的迭代请求,转发给对象内部持有的_children属性上(列表) """
        return iter(self._children)

# 示例
root = Node(0)
child1 = Node(1)
child2 = Node(2)
root.add_children(child1)
root.add_children(child2)
# 1、for会请求到__iter__()方法;2、迭代实例root, 实际上是在迭代root._children属性
for ch in root: 
    print(ch)
Outs: 
Node(1)
Node(2)
总结:

1、Python的迭代协议要求__iter__()返回一个特殊的迭代器对象,由该对象实现的__next__()方法来完成实际的迭代;
2、如果要做的只是迭代另一个容器中的内容,所要做的就是转发迭代请求。

    原文作者:cook__
    原文地址: https://www.jianshu.com/p/fb8a2cc2a770
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞