Проблема с List comprehension

Question

[darke3v]

Задумка такова, нужно через каждые три символа из к переменной secretkey дописывалось число из id.
Получилось так:

secretkey='abcdef'
id='12'
test=[secretkey[i:i + 3]+l for i in range(0, len(secretkey), 3) for l in id]
['abc1', 'abc2', 'def1', 'def2']

Как избавиться от второго цикла дабы избежать дубляжа?

Comments

[darke3v]

В итоге желаю получить
['abc1', 'def2']

Answer 1

[iley]

Можно решить, например, так:

from itertools import cycle
test = [secretkey[i:i + 3]+l for i, l in zip(range(0, len(secretkey), 3), cycle(id))]

Но, на мой взгляд, код будет более читаемым если добавить вспомогательную функцию, разбивающую строку (или любую другую последовательность) на равные части:

from itertools import cycle

def chunks(l, n):
    """ Yield successive n-sized chunks from l."""
    for i in xrange(0, len(l), n):
        yield l[i:i+n]

test = [ chunk + number
         for chunk, number in zip(chunks(secretkey, 3), cycle(id)) ]

Comments

[darke3v]

Идеально, благодарю.