基于键将字典值组合在一起答案

【问题标题】：Group dictionary values together based on a key基于键将字典值组合在一起
【发布时间】：2021-02-18 12:50:14
【问题描述】：

我希望将输入集中的值与作为键的元组中的第一个元素一起分组。第二个元素需要根据公共键组合到一个列表中。输出需要是一个包含元组的列表。

# Input set
values = {(304008, 2020.0), (304008, 2017.0), (250128, 2020.0), (93646, 2020.0), (93646, 2017.0)}

# Current workflow
keys = {i[0] for i in values}

id_dict = dict()
for k in keys:
    id_dict[k] = [int(i[1]) for i in values if i[0] == k]

lst2 = list(id_dict.items())
# Expected output
# [(250128, [2020]), (304008, [2017, 2020]), (93646, [2020, 2017])]

我有预期的输出，但是整个过程太慢了。我希望让它更快。我正在查看 groupby 函数，但我似乎无法让它们工作。

【问题讨论】：

标签： python dictionary itertools

【解决方案1】：

您可以使用itertools.groupby 来完成此操作。基本上按元组中的第一个元素分组，然后列出每个组中的第二个元素。

>>> from itertools import groupby
>>> [(k, [i[1] for i in g]) for k, g in groupby(sorted(values), key=lambda i: i[0])]
[(93646, [2017.0, 2020.0]), (250128, [2020.0]), (304008, [2017.0, 2020.0])]

【讨论】：

【解决方案2】：

您可以使用 setdefault 将键作为元组的第一项创建一个字典，并在集合上进行迭代以单次填充它。使用列表构造函数来获取所需的列表。见下文：

>>> values = {(304008, 2020.0), (304008, 2017.0), (250128, 2020.0), (93646, 2020.0), (93646, 2017.0)}
>>> info = {}
>>> for elements in values:
...     info.setdefault(elements[0], []).append(elements[1])
... 
>>> list(info.items())
[(304008, [2017.0, 2020.0]), (93646, [2017.0, 2020.0]), (250128, [2020.0])]
>>>

这不使用 groupby 但避免了您的第二个循环。

【讨论】：