如何使Python班級覺得

為了讓Python 類可迭代，需實現\_\_iter\_\_ 和\_\_next\_\_ 方法。 1. 實現\_\_iter\_\_ 返回一個迭代器對象（通常為自身或新對象）；2. 在迭代器中定義\_\_next\_\_ 方法控制每次返回值及停止條件；3. 可將迭代器單獨成類以支持多個獨立循環；4. 注意拋出StopIteration 避免無限循環，並確保狀態正確管理。

在Python 中，如果你希望一個類是可迭代的（也就是可以用for循環遍歷），你需要做的是讓這個類支持迭代器協議。這並不復雜，但需要理解兩個關鍵方法： __iter__和__next__ 。

實現__iter__方法

為了讓一個類成為可迭代對象，它必須實現__iter__方法，並返回一個具有__next__方法的對象（通常是它自己，或者一個新的迭代器對象）。

舉個例子，假設你有一個存儲數字的類，你想讓它可以按順序一個個返回這些數字：

 class MyNumbers:
    def __init__(self, start, end):
        self.start = start
        self.end = end
        self.current = start

    def __iter__(self):
        return self

上面這段代碼中， __iter__返回了self ，說明這個類本身也充當了迭代器角色。接下來只需要加上__next__方法。

添加__next__方法控制迭代邏輯

__next__方法負責定義每次迭代返回的值，以及什麼時候停止迭代：

 def __next__(self):
        if self.current > self.end:
            raise StopIteration
        value = self.current
        self.current = 1
        return value

這樣，當你用for遍歷這個類的實例時，就能正常工作了：

 nums = MyNumbers(1, 5)
for num in nums:
    print(num)

輸出將是：

 1
2
3
4
5

分離迭代器與可迭代對象（更靈活的方式）

有時候你不希望自己這個類同時扮演“可迭代對象”和“迭代器”的角色。比如多個循環要獨立進行的時候，就需要每次返回一個新的迭代器。

這時你可以把迭代器單獨做成一個類：

 class MyNumbers:
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def __iter__(self):
        return MyNumbersIterator(self.start, self.end)


class MyNumbersIterator:
    def __init__(self, start, end):
        self.current = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.current > self.end:
            raise StopIteration
        value = self.current
        self.current = 1
        return value

這種方式的好處是每個迭代都是獨立的，不會互相干擾。比如你可以多次調用iter()獲取新的迭代器。

常見問題和注意事項

• 忘記拋出StopIteration異常會導致無限循環。
• 如果你的類實現了__iter__但沒有__next__ ，那它只是一個可迭代對象，而不是迭代器。
• 有時你會看到使用生成器函數（如yield ）來簡化迭代邏輯，但對於自定義類來說，還是推薦顯式實現這兩個方法。
• 如果你只想讓類支持for遍歷，不需要修改狀態，也可以直接在__iter__中返回一個生成器表達式或列表迭代器。

基本上就這些。實現一個可迭代的類不難，只要記住__iter__和__next__是關鍵。寫的時候注意邊界條件和狀態管理，就不會出錯。

以上是如何使Python班級覺得的詳細內容。更多資訊請關注PHP中文網其他相關文章！