Python中的无限生成器，每周几天

Question

我已经看到了类似的问题，我的实用性稍微提高一点，我想一遍又一遍地重复一周的工作日。

到目前为止，我的迭代器不是循环的，请帮我解决这个问题。

def day_generator():
    for w in ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']:
        yield w;

g = day_generator()
print g.next() 

Answer 1

您可以使用itertool的循环： https ：//docs.python.org/2/library/itertools.html#itertools.cycle

import itertools
def day_generator():
    days = ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']
    for day in itertools.cycle(days):
        yield day

长话短说（并在评论中提到）它真的足以使：

day_generator = itertools.cycle(days)

谢谢@FlavianHautbois

Answer 2

你几乎拥有它，你只需要将你的“yield”语句放在一个无限循环中，这样它就会在需要时总是环绕：

def day_generator():
    while True:
        for w in ['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday']:
            yield w

g = day_generator()
for _ in range(10):
    print(next(g))

##Output:
##
##    Monday
##    Tuesday
##    Wednesday
##    Thursday
##    Friday
##    Saturday
##    Sunday
##    Monday
##    Tuesday
##    Wednesday

但是，正如其他人所说， itertools.cycle是最简洁的方法。

Answer 3

itertools.cycle完全符合您的要求：

import itertools

day_generator = itertools.cycle(['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday', 'Saturday', 'Sunday'])