Python的方式來者皆

Question

組列表我有一個看起來像這樣的列表：

[ 
    [1, "a", 319], 
    [1, "b", 991], 
    [2, "Cd", 19], 
    [3, "88x", 2891], 
    [3, "foo", 11] 
] 

是什麼把這個變成最「Python化」的方式：

[ 
    [1, [["a", 319], ["b", 991]]], 
    [2, [["Cd", 19]]], 
    [3, [["88x", 2891], ["foo", 11]]] 
] 

我知道如何使用for循環等來做到這一點，但我正在採用更簡潔的方法。

Answer 1

下面是一個選項假設列表是由第一個元素的前手排序：

from itertools import groupby  
[[k, [x[1:] for x in g]] for k, g in groupby(lst, key = lambda x: x[0])] 

#[[1, [['a', 319], ['b', 991]]], 
# [2, [['Cd', 19]]], 
# [3, [['88x', 2891], ['foo', 11]]]] 

Answer 2

你也可以使用一個collections.defaultdict，如果你不介意一個類似於字典的對象：

>>> l = [ 
... [1, "a", 319], 
... [1, "b", 991], 
... [2, "Cd", 19], 
... [3, "88x", 2891], 
... [3, "foo", 11] 
... ] 
>>> import collections 
>>> d = collections.defaultdict(list) 
>>> for k, *v in l: 
...  d[k].append(v) 
... 
>>> import pprint 
>>> pprint.pprint(d) 
{1: [['a', 319], ['b', 991]], 
2: [['Cd', 19]], 
3: [['88x', 2891], ['foo', 11]]}