Python面向對象編程-魔術方法-__iter__和__next__方法（一） 當前關注

(資料圖片)

在Python面向對象編程中，魔術方法是一種特殊的方法，其名稱以雙下劃線開頭和結尾。這些方法在類定義中使用，用于定義對象的行為。其中，__iter__和__next__方法是用于定義迭代器對象的魔術方法，它們在for循環中起到重要作用。

__iter__方法是迭代器對象的入口點，用于返回一個迭代器對象。它必須返回一個實現了__next__方法的對象，否則會引發TypeError異常。在調用for循環時，Python會自動調用對象的__iter__方法獲取迭代器對象，并依次調用其__next__方法獲取每個元素，直到所有元素都被遍歷完畢。

__next__方法用于返回迭代器中的下一個元素。如果沒有下一個元素，則應該引發StopIteration異常。在每次調用__next__方法時，迭代器對象應該更新其內部狀態，以便在下一次調用__next__方法時返回下一個元素。

下面是一個使用__iter__和__next__方法的例子，展示了如何實現一個簡單的迭代器對象，用于遍歷一個數字范圍內的所有奇數：

class OddIterator:    def __init__(self, start, stop):        self.start = start if start % 2 == 1 else start + 1        self.stop = stop if stop % 2 == 1 else stop - 1        self.current = self.start        def __iter__(self):        return self        def __next__(self):        if self.current > self.stop:            raise StopIteration        result = self.current        self.current += 2        return resultit = OddIterator(1, 10)# 使用for循環遍歷迭代器中的所有元素for x in it:    print(x)  # 輸出: 1 3 5 7 9

在上面的例子中，我們實現了一個OddIterator類，用于遍歷一個數字范圍內的所有奇數。在__init__方法中，我們初始化了迭代器的起始值、結束值和當前值。在__iter__方法中，我們直接返回了迭代器對象自身，因為OddIterator類本身就是一個迭代器對象。在__next__方法中，我們先判斷是否已經遍歷到了范圍內的最后一個奇數，如果是，則引發StopIteration異常。否則，我們返回當前值，并將當前值增加2，以便下一次調用__next__方法返回下一個奇數。

需要注意的是，在實現__next__方法時，我們需要將當前值更新為下一個奇數，以便下一次調用__next__方法能夠正確返回下一個元素。此外，在for循環中遍歷迭代器對象時，如果迭代器已經遍歷到了最后一個元素，則會自動捕獲StopIteration異常，并終止循環。

關鍵詞：