python – 我可以在列表(或字典)理解中对元素进行分组/聚合吗？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 我可以在列表(或字典)理解中对元素进行分组/聚合吗？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1095字，纯文字阅读大概需要2分钟。

内容图文

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

[
    (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 comprehension.

解决方法:

您可以使用collections.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'])})

现在迭代dic：

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

内容总结

以上是互联网集市为您收集整理的python – 我可以在列表(或字典)理解中对元素进行分组/聚合吗？全部内容，希望文章能够帮你解决python – 我可以在列表(或字典)理解中对元素进行分组/聚合吗？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端