代理迭代

問題

你構建了一個自定義容器對象，里面包含有列表、元組或其他可迭代對象。你想直接在你的這個新容器對象上執行迭代操作。

解法

實際上你只需要定義一個 iter () 方法，將迭代操作代理到容器內部的對象上去。比如：

class Node:def __init__(self, value):self._value = valueself._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__() 原理是一樣的。

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.2）：代理迭代的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。