python - 来回循环 Python

Question

我想创建一个无限循环，从 0 到 100 再到 0(依此类推)，只有在满足循环内的某些收敛标准时才会停止，所以基本上是这样的:

for i in range(0, infinity):
    for j in range(0, 100, 1):
        print(j) # (in my case 100 lines of code)
    for j in range(100, 0, -1):
        print(j) # (same 100 lines of code as above)

有没有办法将 j 上的两个 for 循环合并为一个，这样我就不必在循环内写出相同的代码两次？

Answer 1

使用itertools

的 chain方法

import itertools
for i in range(0, infinity):
    for j in itertools.chain(range(0, 100, 1), range(100, 0, -1)):
        print(j) # (in my case 100 lines of code)

正如@Chepner 所建议的，您可以将 itertools.cycle() 用于无限循环:

from itertools import cycle, chain

for i in cycle(chain(range(0, 100, 1), range(100, 0, -1))):
    ....