python - 在不使用生成器函数的情况下，如何在 Python 中实现与 itertools.count 等效的迭代器？

Question

使用生成器函数，可以这样实现itertools.count(来自documentation):

def count(start=0, step=1):
    # count(10) --> 10 11 12 13 14 ...
    # count(2.5, 0.5) -> 2.5 3.0 3.5 ...
    n = start
    while True:
        yield n
        n += step

我正在尝试寻找如何在没有生成器函数的情况下实现类似的迭代器。

class Count:
    def __init__(self, start=0, step=1):
        self.c = start
        self.step = step

    def __iter__(self):
        return self

    def __next__(self):
        n = self.c
        self.c += self.step
        return n

这是实现吗？

当然，它没有任何实际意义，重点只是了解如何使用生成器实现它。

Answer 1

您可以使用collections.abc.Iterator以更少的代码实现计数

from collections.abc import Iterator

class count(Iterator):
    def __init__(self, start=0, step=1):
        self.c, self.step = start-step, step

    def __next__(self):
        self.c += self.step
        return self.c