如何在具有多个匹配值的列表中对元组进行分组

问题描述

我想尽可能有效地将具有多个公共值的元组列表分组。我知道这样做的方法很长，但是想知道这样做会是一种更快，更“ Pythonic”的方法。

我要实现的目标：

lst = [('a','1',1),('a',2),'2',('b',('c',1)]

result = [
        [('a',2)],[('a',1)],[('b',1)]
        [('c',1)]
    ]

理想情况下，我想要得到的真实结果是：

result = [
        [('a',]

如果没有匹配对，则将它们排除在外，但我认为我可以使用set()

解决方法

使用dict.setdefault

例如：

r = {}
for i in lst:
    r.setdefault((i[0],i[1]),[]).append(i)   #key -> First 2 values

print(list(r.values()))
#OR print([i for i in r.values() if len(i) > 1])

输出：

[[('a','1',1),('a',2)],[('a','2',[('b',1)],[('c',1)]]