python – 如何指定itertools.cycle函数的起始位置

我需要在列表中循环开始1-4之间的位置

使用itertools我能够在列表中循环

positions = itertools.cycle([1,2,3,4])
next(positions)

这确实会返回下一个位置,但下次我需要从3开始怎么办?如何设置起始位置?

我需要经常更改起始位置,我不能将列表更改为从3开始.

最佳答案 你不能设置一个起始位置;它总是从给定序列开始的地方开始.

您可以将循环移动几步,然后再根据需要使用它.使用itertools.islice()跳过一些项目:

from itertools import islice

starting_at_three = islice(positions, 2, None)

你传入iterable,然后是一个开始和结束值;这里没有意味着islice()迭代器永远持续或直到底层位置迭代器耗尽.

演示:

>>> from itertools import islice, cycle
>>> positions = cycle([1, 2, 3, 4])
>>> starting_at_three = islice(positions, 2, None)
>>> next(starting_at_three)
3
>>> next(starting_at_three)
4
>>> next(starting_at_three)
1

另一种选择是传递不同的顺序;你可以传入[3,4,1,2].

点赞