当前位置：首页 > news >正文

python自定义迭代器

news 2026/3/27 3:17:50

python自定义迭代器

 1 # 自定义迭代器
 2 class IntegerIterator:
 3     def __init__(self, start: int, end: int):
 4         self.current = start
 5         self.end = end
 6 
 7     def __iter__(self):
 8         return self
 9 
10     def __next__(self):
11         if self.current < self.end:
12             res = self.current
13             self.current += 1
14             return res
15         else:
16             raise StopIteration
17 
18 
19 integerIterator = IntegerIterator(0, 10)
20 for n in integerIterator:
21     print(n, end=' ')