美文网首页
python3学习笔记:委托迭代

python3学习笔记:委托迭代

作者: 潼潼夏 | 来源:发表于2020-02-26 16:59 被阅读0次

    问题

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

    解决方案

    一般来说,定义一个__iter__()方法,将迭代请求委托到对象内部持有的容器上。示例如下:

    class Node:
        def __init__(self, value):
            self._value = value
            self._children = []
    
        def __repr__(self):
            return 'Node({!r})'.format(slef._value)
    
        def add_child(self, node):
            self._children.append(node)
    
        def __iter__(self):
            return iter(self._children)
    
    #Example
    if __name__ == '__main__':
        root = Node(0)
        child1 = Node(1)
        child2 = Node(2)
        root.add_child(child1)
        root.add_child(child2)
        #outputs Node(1) Node(2)
        for ch in root:
            print (ch._value)
    #1
    #2
    

    在上述例子中,__iter__()方法只是简单地将迭代请求转发给对象内部持有的_children属性上。

    python的迭代协议要求__iter__()返回一个特殊的迭代器对象,由该对象实现的next()方法来完成实际的迭代。如果要做的只是迭代另一个容器中的内容,我们不必担心底层细节是如何工作的,所要做的就是转发迭代请求。
    示例中用到的iter()函数对代码做了一定程度的简化。iter(s)通过调用s.__iter__()来简单地返回底层的迭代器,这和len(s)调用s.__len__()的方法是一样的。

    相关文章

      网友评论

          本文标题:python3学习笔记:委托迭代

          本文链接:https://www.haomeiwen.com/subject/akubchtx.html