Списъкът от кортежи на Python обединява 2-ри елемент с уникален първи елемент

Даден списък от кортежи като този:

a = [ ( "x", 1, ), ( "x", 2, ), ( "y", 1, ), ( "y", 3, ), ( "y", 4, ) ]

Какъв би бил най-лесният начин за филтриране за уникален първи елемент и обединяване на втория елемент. Подобен резултат би бил желан.

b = [ ( "x", 1, 2 ), ( "y", 1, 3, 4 ) ]

Благодаря,


person chuy08    schedule 20.09.2013    source източник


Отговори (4)


Можете да използвате defaultdict:

>>> from collections import defaultdict
>>> d = defaultdict(tuple)
>>> a = [('x', 1), ('x', 2), ('y', 1), ('y', 3), ('y', 4)]
>>> for tup in a:
...     d[tup[0]] += (tup[1],)
...
>>> [tuple(x for y in i for x in y) for i in d.items()]
[('y', 1, 3, 4), ('x', 1, 2)]
person TerryA    schedule 20.09.2013

Ето какво измислих:

[tuple(list(el) + [q[1] for q in a if q[0]==el]) for el in set([q[0] for q in a])]
person sashkello    schedule 20.09.2013

В допълнение към предишните отговори, още един ред:

>>> a = [ ( "x", 1, ), ( "x", 2, ), ( "y", 1, ), ( "y", 3, ), ( "y", 4, ) ]
>>> from itertools import groupby
>>> [(key,) + tuple(elem for _, elem in group) for key, group in groupby(a, lambda pair: pair[0])]
[('x', 1, 2), ('y', 1, 3, 4)]
person fjarri    schedule 20.09.2013

Един от начините е чрез използване на израз за разбиране на списък с itertools.groupby , itertools.chain и operator.itemgetter като:

>>> from itertools import groupby, chain
>>> from operator import itemgetter

>>> my_list = [ ( "x", 1, ), ( "x", 2, ), ( "y", 1, ), ( "y", 3, ), ( "y", 4, ) ]

>>> [set(chain(*i)) for _, i in groupby(sorted(my_list), key=itemgetter(0))]
[set(['x', 2, 1]), set(['y', 1, 3, 4])]

Забележка: set са неподредени по природа, така че няма да запазят позицията на елементите. Не използвайте set, ако позицията има значение.

person Anonymous    schedule 18.11.2017