复制可迭代对象的 Pythonic 方式

Question

对于我正在处理的一个小项目，我需要循环浏览一个列表。对于这个循环的每个元素，我必须通过同一个列表开始另一个循环，前一个元素作为新循环的第一个元素。例如，我希望能够产生这样的东西:

1, 2, 3, 4, 1, 2, 3, 4, 1, ...
2, 3, 4, 1, 2, 3, 4, 1, 2, ...
3, 4, 1, 2, 3, 4, 1, 2, 3, ...
4, 1, 2, 3, 4, 1, 2, 3, 4, ...
1, 2, 3, 4, 1, 2, 3, 4, 1, ...
...

我认为在每个 .next() 之后复制一个 itertools.cycle 会保留当前状态，这样我就可以使用来自“外部”循环的元素开始新的循环。或者甚至将循环指针“重置”到较旧的位置。我尝试了以下方法:

>>> import itertools, copy
>>> a = itertools.cycle([1, 2, 3, 4])
>>> b = copy.copy(a)

但是出现了这个错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/lib/python2.6/copy.py", line 95, in copy
    return _reconstruct(x, rv, 0)
  File "/usr/lib/python2.6/copy.py", line 323, in _reconstruct
    y = callable(*args)
  File "/usr/lib/python2.6/copy_reg.py", line 93, in __newobj__
    return cls.__new__(cls, *args)
TypeError: cycle expected 1 arguments, got 0

我知道有很多不同的方法可以实现我想要的，但我正在寻找一些简短、清晰的 pythonic 代码。也许有人有其他想法甚至是片段？事实上它是 not possible to copy iterator objects唤醒了我的兴趣。在需要可迭代对象副本的情况下是否有最佳实践？还是复制可迭代对象通常是愚蠢且无用的？

Answer 1

Is there a best-practice in situations where one wants a copy of an iterable?

itertools.tee 为您提供了两个迭代器，每个迭代器都产生与原始迭代器相同的项目，但它采用原始迭代器并记住它产生的所有内容，因此您不能再使用原始迭代器。不过，它在这里无济于事，因为它会一直记住这些循环值，直到出现 MemoryError。

Or is copying iterables silly and useless in general?

迭代器只是定义为具有当前状态并产生一个项目。您无法判断他们将来是否会生产相同的元素，或者他们过去生产过哪些元素。一个真正的副本必须做到这两点，所以这是不可能的!

在您的情况下，制作一个新循环是如此微不足道，我宁愿这样做也不愿尝试复制现有循环。例如:

def new_cycle( seq, last=None):
    if last is None:
        return cycle(seq)
    else:
        it = cycle(seq)
        while next(it) != last:
            pass
        return it