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

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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)]
另一答案
>>> a = [("x", 1,), ("x", 2,), ("y", 1,), ("y", 3,), ("y", 4,)]
>>> d = {}
>>> for k, v in a:
...     d.setdefault(k, [k]).append(v)
>>> b = map(tuple, d.values())
>>> b
[('y', 1, 3, 4), ('x', 1, 2)]
另一答案

这就是我想出的:

[tuple(list(el) + [q[1] for q in a if q[0]==el]) for el in set([q[0] for q in a])]
另一答案

除了之前的答案,另一个单行:

>>> 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)]
另一答案

一种方法是使用itertools.groupbyitertools.chainoperator.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

以上是关于Python元组列表将第二个元素与唯一的第一个元素合并的主要内容,如果未能解决你的问题,请参考以下文章

按元组的第二个元素对元组列表进行排序,无需高阶函数或递归

将 char 添加到列表中的每个元组的第二个元素

比较列表haskell中的所有元素[重复]

通过将第一个列表中的第一行项与第二个列表中的项进行比较来创建新列表

Python - 遍历没有最后一个元素的列表

如何提取每个子列表中每个元组的第一个元素?