python - 如何无限使用 yield list 元素

Question

我有目录列表，这个元素应该一次返回一个

所以我编码如下:

def gen():
    list = ["dir_1", "dir_2", ..., "dir_n"]
    for i in list:
        yield i

但是，我想一遍又一遍地获取目录，例如 dir_1, dir_2, ... ,dir_n, dir_1 , dir_2... 就像那样。

我该怎么办？

Answer 1

您可以使用循环，不需要另一个生成器，只需在迭代器上调用循环，它将永远循环

from itertools import cycle

for dir in cycle(["dir_1", "dir_2", "dir_n"]):
   do stuff ...
   if some_reason:
     break