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])]