python - 可使用 yield 或 __next__() 迭代

Question

我正在研究制作可迭代对象，我想知道这两个选项中的哪一个更像 pythonic/更好的方法，是没有区别还是我对使用 yield 的想法有误？对我来说，使用 yield 似乎更干净，而且显然它比使用 __next__() 更快，但我不确定。

class iterable_class():

    def __init__(self, n):
        self.i = 0
        self.n = n

    def __iter__(self):
        return self

    def __next__(self):
        if self.i < self.n:
            i = self.i
            self.i += 1
            return i
        else:
            raise StopIteration()

使用 yield :

class iterable_class_with_generator():

    def __init__(self, n):
        self.i = 0
        self.n = n

    def __iter__(self):
        while self.i < self.n:
            yield self.i
            self.i += 1

Answer 1

一个明显的区别是第一个版本实现了一个迭代器(一个具有__next__ 并且其__iter__ 返回自身的对象)，而第二个实现了一个 iterable(一个可以实现 __iter__ 来创建一些迭代器的对象)。在大多数情况下，这没有什么区别，因为 for 语句和所有 itertools 都接受任何可迭代对象。

通过以下代码可以看出区别:

>>> x = iterable_class(10)
>>> next(x)
0
>>> next(x)
1
>>> list(x)
[2, 3, 4, 5, 6, 7, 8, 9]

显然这不适用于 iterable_class_with_generator，因为它没有实现 __next__。但有一个更深层次的区别:因为 list(x) 接受任何可迭代对象，它会首先调用 x.__iter__()，如果是 iterable_class_with_generator 创建一个新的生成器，它将从头开始计数。一个真正的基于生成器的迭代器出现在答案的末尾，但在大多数情况下，差异无关紧要。

关于是使用生成器还是定义你自己的 __next__ 的风格差异，两者都将被识别为正确的 Python，因此你应该选择对将要编写的人或团队来说更好读的那个维护代码。由于生成器版本较短，而且生成器是一种易于理解的 Python 惯用语，因此我会选择那个。

请注意，如果您使用生成器实现 __iter__，则无需在实例变量中保留迭代状态，因为生成器会为您完成。代码就更简单了:

class iterable_class_with_generator:
    def __init__(self, n):
        self.n = n

    def __iter__(self):
        for i in range(self.n):
            yield i

---

最后，这是 iterable_class_with_generator 的一个版本，它实现了一个在内部使用生成器的真正的迭代器:

class iterable_class_with_generator:
    def __init__(self, n):
        self._gen = self._generate(n)

    def __iter__(self):
        return self

    def __next__(self):
        return next(self._gen)

    def _generate(self, n):
        for i in range(n):
            yield i