python - Python中的过滤操作

Question

我正在寻找一种无需太多样板代码即可在 Python 中执行此操作的方法。

假设我有一个列表:

[(a,4),(b,4),(a,5),(b,3)]

我正在尝试找到一个函数，允许我按第一个元组值排序，并将列表值合并在一起，如下所示:

[(a,[4,3]),(b,[4,5])]

我知道我可以用天真的方法来做到这一点，但我想知道是否有更好的方法。

Answer 1

使用collections.defaultdict(list):

from collections import defaultdict

lst = [("a",4), ("b",4), ("a",5), ("b",3)]

result = defaultdict(list)
for a, b in lst:
    result[a].append(b)

print sorted(result.items())

# prints: [('a', [4, 5]), ('b', [4, 3])]

排序前算法的复杂度为O(n)；按算法分组有 O(n * log(n)) 并且 set/list/dict 理解有大于 O(n^2)