溫馨提示×

Python迭代器協(xié)議是否支持自定義對象

小樊
82
2024-11-01 23:23:08
欄目: 編程語言

是的,Python的迭代器協(xié)議支持自定義對象。要實現(xiàn)一個自定義對象的迭代器,你需要定義兩個方法:__iter__()__next__()。__iter__() 方法返回迭代器對象本身,而 __next__() 方法返回容器的下一個值。當容器中沒有更多元素時,__next__() 方法應(yīng)該拋出一個 StopIteration 異常。

以下是一個簡單的自定義迭代器示例,用于計算一個數(shù)字列表中所有偶數(shù)的和:

class EvenSumIterator:
    def __init__(self, numbers):
        self.numbers = numbers
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        while self.index < len(self.numbers):
            if self.numbers[self.index] % 2 == 0:
                sum_even_numbers += self.numbers[self.index]
                self.index += 1
                return self.numbers[self.index - 1]
            self.index += 1
        raise StopIteration

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9]
even_sum_iterator = EvenSumIterator(numbers)

for even_number in even_sum_iterator:
    print(even_number)

在這個示例中,我們創(chuàng)建了一個名為 EvenSumIterator 的自定義迭代器類,它接受一個數(shù)字列表作為輸入。我們實現(xiàn)了 __iter__()__next__() 方法,以便在迭代過程中計算所有偶數(shù)的和。當我們遍歷 even_sum_iterator 時,它將輸出列表中的所有偶數(shù)。

0