Python：按第一个元素连接列表列表

Question

我正在尝试通过列表中的第一个元素组合列表/元组列表-像这样：

输入：

[(1, [32, 432, 54]), (1, [43, 54, 65]), (2, [2, 43, 54]), (2, [1, 5, 6])]

输出：

[(1, [32, 432, 54], [43, 54, 65]), (2, [2, 43, 54], [1, 5, 6])]

列表实际上是按第一个元素排序的，就像在我的示例输入中一样，并且最后元组是否是列表也没有关系。

有没有一种有效的/ python方式来做到这一点？

Answer 1

使用itertools.groupby和list comprehension ：

>>> lst = [(1, [32, 432, 54]), (1, [43, 54, 65]), (2, [2, 43, 54]), (2, [1, 5, 6])]
>>> import itertools
>>> [(key,) + tuple(v for k, v in grp)
...     for key, grp in itertools.groupby(lst, key=lambda x: x[0])]
[(1, [32, 432, 54], [43, 54, 65]), (2, [2, 43, 54], [1, 5, 6])]