如何按每对的第二个元素对 Python 中的对列表进行有效分组,同时将第一个元素保留为分组结果中的列表?

Mary-Kate Olsen
发布: 2024-10-31 04:11:01
原创
254 人浏览过

How can I efficiently group a list of pairs in Python by the second element of each pair, while retaining the first element as a list within the grouped result?

Python Group By

人们可能会遇到需要按每对中的第二个元素对一组数据对进行分组,同时将第一个元素保留为列表的情况分组结果。这可以在 Python 中使用以下步骤有效地实现。

使用字典

使用集合模块中的 defaultdict 创建一个字典,其中键是该对的第二个元素。然后,迭代输入列表并将第一个元素附加到相应键的值。

<code class="python">import collections

input = [
    ('11013331', 'KAT'),
    ('9085267', 'NOT'),
    ('5238761', 'ETH'),
    ('5349618', 'ETH'),
    ('11788544', 'NOT'),
    ('962142', 'ETH'),
    ('7795297', 'ETH'),
    ('7341464', 'ETH'),
    ('9843236', 'KAT'),
    ('5594916', 'ETH'),
    ('1550003', 'ETH'),
]

res = collections.defaultdict(list)
for v, k in input:
    res[k].append(v)</code>
登录后复制

使用列表理解将字典转换为预期的 JSON 格式:

<code class="python">result = [{'type': k, 'items': v} for k, v in res.items()]</code>
登录后复制

使用itertools.groupby

另一种方法涉及使用 itertools.groupby,但要求输入列表按第二个元素排序。

<code class="python">from operator import itemgetter
from itertools import groupby

sorted_input = sorted(input, key=itemgetter(1))
groups = groupby(sorted_input, key=itemgetter(1))</code>
登录后复制

使用推导式创建字典列表:

<code class="python">result = [{'type': k, 'items': [x[0] for x in v]} for k, v in groups]</code>
登录后复制

Python 版本注意事项

Python 3.7 之前,不保留字典中键的顺序。要保持原始顺序,请使用 collections.OrderedDict。

<code class="python">from collections import OrderedDict

res = OrderedDict()
for v, k in input:
    if k in res:
        res[k].append(v)
    else:
        res[k] = [v]</code>
登录后复制

从 Python 3.7 开始,常规字典会保持插入顺序,因此不再需要 OrderedDict。

以上是如何按每对的第二个元素对 Python 中的对列表进行有效分组,同时将第一个元素保留为分组结果中的列表?的详细内容。更多信息请关注PHP中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责声明 Sitemap
PHP中文网:公益在线PHP培训,帮助PHP学习者快速成长!