我需要在列表中循环开始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].