python - 为什么 __next__() 内部的 yield 会返回生成器对象？

Question

我正在使用 yield 在我的类中的 __next__() 函数中返回下一个值。但是它不返回下一个值，它返回生成器对象。

我正在尝试更好地理解迭代器和 yield。我可能做错了。

看看。

class MyString:
    def __init__(self,s):
        self.s=s

    def __iter__(self):
        return self

    def __next__(self):
        for i in range(len(self.s)):
            yield(self.s[i])

r=MyString("abc")
i=iter(r)
print(next(i))

返回:

生成器对象 __next__ 位于 0x032C05A0

Answer 1

next 在这种情况下几乎只是调用 __next__()。在你的对象上调用 __next__ 将启动生成器并返回它(此时没有魔法)。

---

在这种情况下，您可能根本不定义 __next__ 就可以逃脱:

class MyString:
    def __init__(self,s):
        self.s=s

    def __iter__(self):
        for i in range(len(self.s)):
            yield(self.s[i])
        # Or...
        # for item in self.s:
        #     yield item

如果你想使用 __iter__ 和 __next__(定义 iterator 而不是简单地制作 iterable )，你可能想做一些事情像这样:

class MyString:
    def __init__(self,s):
        self.s = s
        self._ix = None

    def __iter__(self):
        return self

    def __next__(self):
        if self._ix is None:
            self._ix = 0

        try:
            item = self.s[self._ix]
        except IndexError:
            # Possibly reset `self._ix`?
            raise StopIteration
        self._ix += 1
        return item