是的,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ù)。