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,None)

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

演示：

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

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