人们可能会遇到需要按每对中的第二个元素对一组数据对进行分组,同时将第一个元素保留为列表的情况分组结果。这可以在 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,但要求输入列表按第二个元素排序。
<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 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中文网其他相关文章!