Python生成器类代码

生成器类是在 Python 中用于生成迭代器的一种方法。它通过实现 __iter__() 和 __next__() 方法来生成迭代器。

下面是一个简单的例子：

class MyGenerator:
    def __init__(self, start, end):
        self.start = start
        self.end = end

    def __iter__(self):
        return self

    def __next__(self):
        if self.start < self.end:
            num = self.start
            self.start += 1
            return num
        raise StopIteration

使用方法：

gen = MyGenerator(0, 10)
for i in gen:
    print(i)

输出：

0
1
2
3
4
5
6
7
8
9