как напечатать этот шаблон, используя итератор python

Question

L = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
head = 'head' 
tail = 'tail' 

Предположим, мы можем и можем получить только итератор некоторого итерируемого (L). и мы не можем знать длину L. Это можно напечатать итерацию как:

'head123tail' 
'head456tail' 
'head789tail' 
'head10tail' 

Мой попробовать на это следующим образом.

L = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
head = 'head' 
tail = 'tail' 
slice_size = 3 

i = iter(L) 
try: 
    while True: 
     counter = 0 
     while counter < slice_size: 
      e = next(i) 
      if counter == 0: 
       print(head, end='') 
      print(e, end='') 
      counter += 1 
     else: 
      print(tail) 
except StopIteration: 
    if counter > 0: 
     print(tail) 

Answer 1

Вот один из способов сделать это с itertools.groupby и itertools.count.

groupby на ключевых функции lambda _: next(c)//3 групп элементов в итерируемых в тройках в последовательностях. Логика использует целочисленное деление следующего объекта в счетный пункт на 3:

from itertools import groupby, count 

L = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
head = 'head' 
tail = 'tail' 

c = count()  
for _, g in groupby(L, lambda _: next(c)//3): 
    item = head + ''.join(map(str, g)) + tail 
    print(item) 

Выход:

head123tail 
head456tail 
head789tail 
head10tail 

Answer 2

Вы можете split the iterator into chunks из трех, используя chain и slice от itertools и for и присоедините их. Цикл for сделает большую часть вашей конструкции try/while True/except.

L = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
head = 'head' 
tail = 'tail' 
slice_size = 3 
iterator = iter(L) 

from itertools import chain, islice 
for first in iterator: 
    chunk = chain([first], islice(iterator, slice_size - 1)) 
    print(head, ''.join(str(x) for x in chunk), tail) 

Однако, если итератор просто list, вы можете просто использовать range с step параметра:

for start in range(0, len(L), slice_size): 
    chunk = L[start : start + slice_size] 
    print(head, ''.join(str(x) for x in chunk), tail)