实现一个可逆迭代类来替换一个生成器

Question

在 python 中，我有一个类，它有一个返回生成器的方法：

class foo():
    data = [1, 2, 3]
    def mygen(self):
        for d in self.data:
            yield d

instance = foo()
print([i for i in instance.mygen()])

但我无法扭转这一点：

print([i for i in reversed(instance.mygen())])
TypeError: 'generator' object is not reversible

所以我想我可以实现一个在调用__iter__ 时返回生成器的类，像这样

class foo():
    data = [1, 2, 3]
    def mygen(self):
        return _ReversibleIterator(self)

class _ReversibleIterator(object):
    def __init__(self, obj):
        self.obj = obj

    def __iter__(self):
        for d in obj.data:
            yield d

    def __reversed__(self):
        for d in reversed(obj.data):
            yield d

但我认为这不太一样，因为_ReversibleIterator 类没有next() 方法。

那么创建返回迭代器的类方法的pythonic方法是什么？可以是reversed()？

（显然我只是以[1,2,3] 为例。真正要迭代的东西是不那么简单可逆的）

Answer 1

根据docs，reversed 必须使用以下两种方法之一：__reversed__ 方法或__len__ 和__getitem__ 方法。如果您考虑一下，这是有道理的，因为大多数生成器不能支持reversed，因为它们会即时生成结果：它们不知道下一个元素是什么，更不用说最后一个元素了成为。但是，如果您知道它的长度并且可以随机访问任何索引，则可以反转它。

class foo():
    data = [1, 2, 3]
    def mygen(self):
        return _ReversibleIterator(self)

class _ReversibleIterator(object):
    def __init__(self, obj):
        self.obj = obj
        self.index = 0

    def __iter__(self):
        self.index = 0
        return self

    def __reversed__(self):
        return reversed(self.obj.data)

    def __next__(self):
        try:
            el = self.obj.data[self.index]
        except IndexError:
            raise StopIteration
        self.index += 1
        return el

或

class _ReversibleIterator(object):
    def __init__(self, obj):
        self.obj = obj
        self.index = 0

    def __iter__(self):
        self.index = 0
        return self

    def __len__(self):
        return len(self.obj.data)

    def __getitem__(self, i):
        return self.obj.data[i]

    def __next__(self):
        try:
            el = self[self.index]
        except IndexError:
            raise StopIteration
        self.index += 1
        return el

顺便说一句，如果您愿意，可以将for d in whatever: yield d 替换为yield from whatever。