温馨提示×

Python迭代器协议是否支持自定义对象

小樊
82
2024-11-01 23:23:08
栏目: 编程语言

是的,Python的迭代器协议支持自定义对象。要实现一个自定义对象的迭代器,你需要定义两个方法:__iter__()__next__()__iter__() 方法返回迭代器对象本身,而 __next__() 方法返回容器的下一个值。当容器中没有更多元素时,__next__() 方法应该抛出一个 StopIteration 异常。

以下是一个简单的自定义迭代器示例,用于计算一个数字列表中所有偶数的和:

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)

在这个示例中,我们创建了一个名为 EvenSumIterator 的自定义迭代器类,它接受一个数字列表作为输入。我们实现了 __iter__()__next__() 方法,以便在迭代过程中计算所有偶数的和。当我们遍历 even_sum_iterator 时,它将输出列表中的所有偶数。

0