我有一个循环:
result = []
list = ['a', 'b', 'c', 'd', 'e', 'f']
start = 2
step = 5
end = start + step*len(list)
for i in range(start, end, step):
result.append(list[i%len(list)])
print result
在这种情况下,结果将是:
['c', 'b', 'a', 'f', 'e', 'd']
但是假设,我希望结果是(将起始索引更改为 1):
['b', 'a', 'd', 'c', 'f', 'e']
如何在每次循环后更改步长,以便在第一个循环中步长为 5,在下一个循环中步长为 3,然后再次为 5,依此类推?
请您参考如下方法:
你可以为这种事情写你自己的生成器:
from itertools import cycle
def super_range(start, stop, steps):
csteps = cycle(steps)
val = start
while val < stop:
yield val
val += next(csteps)
用法看起来像这样:
for i in super_range(start, end, (5, 3)):
result.append(list[i%len(list)])
