Создание нескольких копий элементов списка

Question

У меня есть последовательность элементов, каждая из которых связана с ее весом w (положительное целое число) в кортеже. Что хорошего (предпочтительно, на основе итератора) создать последовательность или список, содержащий w_i экземпляров каждого элемента e_i?

weighted = [ ("a", 3), ("b", 1), ("c", 4) ] 

unrolled = [ "a", "a", "a", "b", "c", "c", "c", "c" ] 

Answer 1

Использования списка понимания с вложенным циклом:

unrolled = [c for c, count in weighted for _ in range(count)] 

Если вы используете Python 2, вы можете использовать вместо xrange().

Если вы хотите itertools, вы можете использовать itertools.chain.from_iterable(), чтобы сделать это в ленивой Iterable:

from itertools import chain 

chain.from_iterable([c] * count for c, count in weighted) 

Демо:

>>> weighted = [ ("a", 3), ("b", 1), ("c", 4) ] 
>>> [c for c, count in weighted for _ in range(count)] 
['a', 'a', 'a', 'b', 'c', 'c', 'c', 'c'] 
>>> from itertools import chain 
>>> list(chain.from_iterable([c] * count for c, count in weighted)) 
['a', 'a', 'a', 'b', 'c', 'c', 'c', 'c'] 

Я использовал list() превратить chain итератор в последовательность.

Answer 2

Этот метод в основном создает одну строку с использованием выражения генератора и join, а затем превращает эту строку в list.

weighted = [ ("a", 3), ("b", 1), ("c", 4) ] 
unrolled = list(''.join([letter*count for letter,count in weighted])) 

>>> unrolled 
['a', 'a', 'a', 'b', 'c', 'c', 'c', 'c'] 

Answer 3

Counter имеет эту логику встроенная, предполагая, что порядок не важен

>>> weighted = [ ("a", 3), ("b", 1), ("c", 4) ] 
>>> from collections import Counter 
>>> c = Counter(dict(weighted)) 
>>> list(c.elements()) 
['a', 'a', 'a', 'c', 'c', 'c', 'c', 'b'] 

Другой способ с itertools (их много)

>>> from itertools import chain, repeat, starmap 
>>> list(chain.from_iterable(starmap(repeat, weighted))) 
['a', 'a', 'a', 'b', 'c', 'c', 'c', 'c']