可以恢复迭代器并且可以分配它的值/状态吗？

Question

我有下面的 sn-p 它使用生成器来提供新的 ID

...

    def __init__(self, id_generator = None):
        if id_generator is None: id_generator = 0

        if isinstance(id_generator, int):
            import itertools
            self._generator = itertools.count(id_generator)

        else:
            self._generator = id_generator
...

    id = self._generator.next()  #0
    id = self._generator.next()  #1
    id = self._generator.next()  #2
    id = self._generator.next()  #3

我的问题是，假设我有一个现有的 id 号码，例如 int(99) 需要被接受 在运行时需要_generator生成从99开始的ID。 看起来像：

0
1
2
3
4
             <---- new id number 99 be given in somehow
99
100
101

如何将它反馈给 _generator？ 只需使用self._generator = itertools.count(99) 还是有更好的方法？

更多，sucn 混淆可以更概括：我如何备忘录/分配生成器的 如果不是 int 类型的状态/值？

谢谢！

Answer 1

不，生成器只是生成项目，一旦创建，您就无法设置或保存它们的状态。所以self._generator = itertools.count(99) 确实是最好的选择。

您可以用itertools.tee 复制一个生成器，它会记住第一个可迭代对象的输出序列并将其传递给新的生成器。

您还可以编写一个从您可以更改的源中提取的生成器：

class counter(object):
    def __init__(self, current=0):
        self.current = current

    def __iter__(self):
        def iter():
            while True:
                yield self.current
                self.current += 1 # 
        return iter()

    def set(self,x):
        self.current = x

s = counter()
t = iter(s)
print t.next() # 0
s.set(20)
print t.next() # 21