在元组列表中使用相同的相同键对所有值进行分组

Question

我有一个包含字符串和字典的元组列表，如下所示：

# data type: List<Tuple<string, dict>>    
input_data_structure = [
('key1', {'a': 'b'}), 
('key2', {'w': 'x'}), 
('key1', {'c': 'd'}), 
('key2', {'y': 'z'})]

我想用相同的键对所有值进行分组。 所以结果可能看起来像这样或类似：

# data type: List<Tuple<string, List<dict>>>     
result_data_structure = [
('key1',  [{'a': 'b'}, {'c': 'd'}]), 
('key2',  [{'w': 'x'}, {'y': 'z'}])]

对我来说，拥有一个良好的数据结构很重要，我可以在其中循环遍历现有的 arrays 键以获得如下值：

for t in result:
     for val in t[1]:
         print(val)

有人知道如何处理或转换数据吗？ 提前致谢！

Answer 1

您可以使用defaultdict轻松实现此目的。

from collections import defaultdict
d = defaultdict(list)

for key, value in input_data_structure:
    d[key].append(value)

d  # defaultdict(<class 'list'>, {'key1': [{'a': 'b'}, {'c': 'd'}], 'key2': [{'w': 'x'}, {'y': 'z'}]})

如果您需要您的 output 作为元组键/值列表，那么您可以执行此行。

list(d.items())  # [('key1', [{'a': 'b'}, {'c': 'd'}]), ('key2', [{'w': 'x'}, {'y': 'z'}])]

Answer 2

没有导入的解决方案：

result = {}
for key, data in input_data_structure:
    try:
        result[key].append(data)
    except KeyError:
        result[key] = [data]
result = list(result.items())

Answer 3

这是使用itertools.groupby的好地方，但请记住，它仅适用于已排序的数据。

from itertools import groupby
from operator import itemgetter

input_data_structure = [
('key1', {'a': 'b'}), 
('key2', {'w': 'x'}), 
('key1', {'c': 'd'}), 
('key2', {'y': 'z'})]

sorted_data = sorted(input_data_structure, key=itemgetter(0))
# [('key1', {'a': 'b'}), ('key1', {'c': 'd'}), ('key2', {'w': 'x'}), ('key2', {'y': 'z'})]
grouped_data = [(k, list(map(itemgetter(1), g))) for k, g in groupby(sorted_data, itemgetter(0))] 
# [('key1', [{'a': 'b'}, {'c': 'd'}]), ('key2', [{'w': 'x'}, {'y': 'z'}])]