是的,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
时,它将输出列表中的所有偶数。