Python元组列表将第二个元素与唯一的第一个元素合并


问题内容

给出如下的元组列表:

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

筛选唯一的第一个元素并合并第二个元素的最简单方法是什么。需要这样的输出。

b = [ ( "x", 1, 2 ), ( "y", 1, 3, 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)]