python - 从零到n到零的单个python循环

Question

是否有一种 Python 式的方法来创建一个遍历从 0 到 n 然后返回到 0 的循环？我可以只创建 2 个循环(一个正向和一个反向)，但我正在寻找单循环。

我的目标是有一个从 0 到 1024 到 0 计数的无限循环。

Answer 1

您可以为此使用 itertools:

如果你想去 1024 并返回一次，你可以使用:

from itertools import <b>chain</b>

for i in <b>chain(</b>range(0,1024),range(1024,0,-1)<b>)</b>:
    print(i)

如果你经常需要这个，你可以使用一个函数来生成可迭代对象:

def range_back(start,end):
    return chain(range(start,end),range(end,start,-1))

并像这样使用它:

for i in <b>range_back(0,1024)</b>:
    print(i)

或者如果你想无限次地这样做:

from itertools import chain<b>, cycle</b>

for i in <b>cycle(</b>chain(range(0,1024),range(1024,0,-1))<b>)</b>:
    print(i)