首頁 > 後端開發 > Python教學 > 如何在 Python 中使用 `__iter__` 和 `__next__` 建立自訂迭代器?

如何在 Python 中使用 `__iter__` 和 `__next__` 建立自訂迭代器?

Patricia Arquette
發布: 2024-12-17 14:56:17
原創
854 人瀏覽過

How to Build Custom Iterators in Python Using `__iter__` and `__next__`?

在Python 中建立基本迭代器

假設您有一個封裝值集合的類,並且您想要建立一個迭代器來允許您按順序存取這些值。

要建立迭代器,請實作迭代器協議,這需要定義兩個方法:

1. __iter__(): 初始化並傳回迭代器物件本身。

2. __next__(): 傳回序列中的下一個值,如果沒有更多值,則引發 StopIteration。

示例迭代器:

考慮以下類的列表值:

class Example:
    def __init__(self, values):
        self.values = values

    # __iter__ returns the iterable itself
    def __iter__(self):
        return self

    # __next__ returns the next value and raises StopIteration
    def __next__(self):
        if len(self.values) > 0:
            return self.values.pop(0)
        else:
            raise StopIteration()
登入後複製

用法:

使用此迭代器,您可以迭代範例類別中的值:

e = Example([1, 2, 3])
for value in e:
    print("The example object contains", value)
登入後複製

This將列印:

The example object contains 1
The example object contains 2
The example object contains 3
登入後複製

迭代器自訂:

如上例所示,迭代器的next 方法可以控制如何取得和傳回值,為自訂迭代器提供更大的靈活性。

以上是如何在 Python 中使用 `__iter__` 和 `__next__` 建立自訂迭代器?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

來源:php.cn
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
作者最新文章
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板