`Itertools.cycle`: самый pythonic способ сделать несколько шагов?

Question

У меня есть список, и я ищу для этого цикл. Беда в том, что меня не интересует циклическое перемещение по элементам по одному, но я хочу циклически перемещаться по элементам n за раз.

Например, если мой список l = ["a", "b", "c", "d"], то я хотел бы следующий вывод:

>>> from itertools import cycle 
>>> gen = cycle(l) 
>>> next(gen, 4) # I know this doesn't work -- take as pseudocode 
d 
>>> next(gen, 3) 
c 

Я знаю, что я могу сделать это с чем-то вроде:

def generator_step(g, n): 
    for item in range(n): 
     next(g) 
    return item 

Answer 1

Вы можете использовать itertools.islice для продвижения объекта цикла до следующего вызова:

from itertools import cycle, islice 

c = cycle(l) 
n = 4 
print(next(islice(c, n-1, n))) 
# d 

Answer 2

В документации для itertools есть handy recipe for precisely this problem.

def consume(iterator, n): 
    "Advance the iterator n-steps ahead. If n is none, consume entirely." 
    # Use functions that consume iterators at C speed. 
    if n is None: 
     # feed the entire iterator into a zero-length deque 
     collections.deque(iterator, maxlen=0) 
    else: 
     # advance to the empty slice starting at position n 
     next(islice(iterator, n, n), None)