发布于 2015-08-30 07:43:05 | 84 次阅读 | 评论: 0 | 来源: 网络整理
实际上你只需要定义一个 __iter__() 方法，将迭代操作代理到容器内部的对象上去。比如：
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)
# 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)
在上面代码中，__iter__() 方法只是简单的将迭代请求传递给内部的 _children 属性。
Python的迭代器协议需要 __iter__() 方法返回一个实现了 __next__() 方法的迭代器对象。
如果你只是迭代遍历其他容器的内容，你无须担心底层是怎样实现的。你所要做的只是传递迭代请求既可。
这里的 iter() 函数的使用简化了代码，
iter(s) 只是简单的通过调用 s.__iter__() 方法来返回对应的迭代器对象，
就跟 len(s) 会调用 s.__len__() 原理是一样的。