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