Python 我可以在列表（或dict）中对元素进行分组/聚合吗？

我有一个元组列表，如下所示：

[
    (1, "red")
    (1, "red,green")
    (1, "green,blue")
    (2, "green")
    (2, "yellow,blue")
]

我正在尝试汇总数据，以便获得以下dict输出：

{
    1: ["red", "green", "blue"]
    2: ["green", "yellow", "blue"]
}

注意事项是：将主键（数字）的颜色字符串组合在一起，然后分割成一个列表，并进行重复数据消除（例如，使用

set

）

我还想做倒数，并按颜色分组：

{
    "red": [1],
    "green": [1, 2]
    "yellow": [2]
    "blue": [1, 2]
}

---

显然，我可以通过循环所有元组来实现这一点，但如果可能的话，我想尝试使用list/dict理解来实现这一点。

您可以使用

集合。defaultdict

：

>>> from collections import defaultdict
>>> lis = [                            
    (1, "red"),
    (1, "red,green"),
    (1, "green,blue"),
    (2, "green"),
    (2, "yellow,blue"),
]
>>> dic = defaultdict(set)       #sets only contain unique items
for k, v in lis:
    dic[k].update(v.split(','))

>>> dic
defaultdict(<type 'set'>,
{1: set(['blue', 'green', 'red']),
 2: set(['blue', 'green', 'yellow'])})

---

使用dict理解生成唯一元素计数的单线解决方案：

X=[1,2,1,3,1,4,1,5,2,3,2,4,2,5,3,4,3,5,4,5,5]

Xagg={xx:sum（[int（y==xx）表示X中的y]）表示集合（X）中的xx}

---

Xagg

欢迎来到SO！试着为你的答案提供更多的解释。请参阅以帮助您改进答案。理解力在哪里？

>>> dic2 = defaultdict(list)
for k,v in dic.iteritems():
    for val in v:
        dic2[val].append(k)
...         
>>> dic2
defaultdict(<type 'list'>,
{'blue': [1, 2],
 'green': [1, 2],
 'yellow': [2],
 'red': [1]})