Python 生成器类代码实现
时间: 2023-05-28 16:05:06 浏览: 102
Python生成器函数
Python 生成器可以通过使用生成器函数或生成器表达式来实现。如果您想要使用类实现生成器,可以使用以下代码:
```
class MyGenerator:
def __init__(self, start, end):
self.start = start
self.end = end
def __iter__(self):
current = self.start
while current < self.end:
yield current
current += 1
gen = MyGenerator(0, 5)
for i in gen:
print(i)
```
这将打印出从0到4的数字。生成器类需要实现 `__iter__` 方法,该方法需要返回一个生成器对象,该生成器对象使用 `yield` 语句生成值。
阅读全文