如何无限迭代列表，每个循环有1个偏移量

Question

我想在列表中从0到结尾无限迭代，但在下一个循环中，我想从1开始到结尾加0，下一个循环将从2开始到结尾加0，1，直到最后一个项目，它将再次从0开始到结尾。

这是我的代码：

a = [ 0, 1, 2 ]
offset = 0
rotate = 0

while True:
    print(a[rotate])
    offset += 1
    rotate += 1
    if offset >= len(a):
        offset = 0
        rotate += 1
    if rotate >= len(a):
        rotate = 0

这就是我目前提出的解决方案。这远非完美。

我想要的结果是：

0, 1, 2 # first iteration
1, 2, 0 # second iteration
2, 0, 1 # third iteration
0, 1, 2 # fourth iteration

等等

Answer 1

您可以使用列表切片创建具有偏移的列表，然后使用itertools无限重复它们。cycle（）。这将精确计算所有旋转一次，然后循环所有旋转：

from itertools import cycle, islice

lst = [0, 1, 2]
concat_lst = lst + lst
items = [concat_lst[i:i+len(lst)] for i in range(len(lst))]
iterator = cycle(items)

for item in islice(iterator, 10):
    print(item)

一旦您通过了预计算，上述方法就很快了，但您可能（取决于您的用例）更喜欢不需要预先花费时间/空间的方法。在这种情况下，可以使用生成器：

from itertools import cycle, islice

def rotate(lst):
    concat_lst = lst + lst
    for start in cycle(range(len(lst))):
        yield concat_lst[start:start+len(lst)]

lst = [0, 1, 2]
for item in islice(rotate(lst), 10):
    print(item)

这两种输出：

[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
[0, 1, 2]

Answer 2

这里有另一种使用指针的方法：

a = [ 0, 1, 2 ]
i = 0
l = len(a)
while True:
  out = []
  for j in range(i, i+l):
    out.append(a[j%l])
  print(out)
  i=(i+1)%l

输出：

[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
[0, 1, 2]
[1, 2, 0]
[2, 0, 1]

Answer 3

尝试以下操作：

a = [0, 1, 2]

while True:
    print(a)
    a.append(a[0])
    a.pop(0)

输出：

[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
[0, 1, 2]
[1, 2, 0]
[2, 0, 1]
...

编辑（Tomerikoo建议）：要删除括号，请使用

a = [0, 1, 2]

while True:
    print(*a, sep=', ')
    a.append(a[0])
    a.pop(0)

输出：

0, 1, 2
1, 2, 0
2, 0, 1
0, 1, 2
1, 2, 0
2, 0, 1
...

编辑2（ShadowRanger建议）：pop返回删除的元素，因此可以简化

a = [0, 1, 2]

while True:
    print(*a, sep=', ')
    a.append(a.pop(0))

输出：

0, 1, 2
1, 2, 0
2, 0, 1
0, 1, 2
1, 2, 0
2, 0, 1
...