list.extend и понимание списка - PullRequest
       6

list.extend и понимание списка

25 голосов
/ 10 октября 2010

Когда мне нужно добавить несколько идентичных элементов в список, я использую list.extend:

a = ['a', 'b', 'c']
a.extend(['d']*3)

Результат

['a', 'b', 'c', 'd', 'd', 'd']

Но как сделать подобное с пониманием списка?

a = [['a',2], ['b',2], ['c',1]]
[[x[0]]*x[1] for x in a]

Результат

[['a', 'a'], ['b', 'b'], ['c']]

Но мне нужен этот

['a', 'a', 'b', 'b', 'c']

Есть идеи?

Ответы [ 6 ]

33 голосов
/ 10 октября 2010

Сложенные LC.

[y for x in a for y in [x[0]] * x[1]]
5 голосов
/ 10 октября 2010

itertools подход:

import itertools

def flatten(it):
    return itertools.chain.from_iterable(it)

pairs = [['a',2], ['b',2], ['c',1]]
flatten(itertools.repeat(item, times) for (item, times) in pairs)
# ['a', 'a', 'b', 'b', 'c']
5 голосов
/ 10 октября 2010
>>> a = [['a',2], ['b',2], ['c',1]]
>>> [i for i, n in a for k in range(n)]
['a', 'a', 'b', 'b', 'c']
3 голосов
/ 10 октября 2010

Если вы предпочитаете расширять списочные представления:

1 голос
/ 10 октября 2010
>>> a = [['a',2], ['b',2], ['c',1]]
>>> sum([[item]*count for item,count in a],[])
['a', 'a', 'b', 'b', 'c']
1 голос
/ 10 октября 2010
import operator
a = [['a',2], ['b',2], ['c',1]]
nums = [[x[0]]*x[1] for x in a]
nums = reduce(operator.add, nums)
...