Python-合并具有相同值的字典键

在处理数据时，有时需要根据字典中的值来合并键，特别是当这些值是集合类型且内容相同时。本文将展示如何使用Python实现这一功能。

给定一个包含字符串键和集合值的字典，目标是编写一个程序来合并那些拥有相同值（不考虑顺序）的键。

示例

输入: test_dict = {'gfg': {5, 4, 3}, 'is': {4, 3, 5}, 'best': {1, 4, 3}, 'for': {1, 3, 4}, 'geeks': {1, 2, 3}}

输出: {'gfg-is': frozenset({3, 4, 5}), 'best-for': frozenset({1, 3, 4}), 'geeks': frozenset({1, 2, 3})}

解释: 具有相同集合值的键被合并，例如 {5, 4, 3} 和 {4, 3, 5} 相等，因此 'gfg' 和 'is' 被合并为 'gfg-is'。

方法一：使用 defaultdict() + join()

此方法通过使用 defaultdict() 来哈希每个集合及其对应的键，然后将具有相同集合值的所有键合并。

# Python3 代码示例
from collections import defaultdict

# 初始化字典
test_dict = {'gfg': {5, 4, 3}, 'is': {4, 3, 5}, 'best': {1, 4, 3}, 'for': {1, 3, 4}, 'geeks': {1, 2, 3}}

# 打印原始字典
print("原始字典: ", test_dict)

# 创建默认字典用于存储哈希值和键
hash_vals = defaultdict(list)

# 遍历字典项，使用 frozenset 哈希值
for key, val in test_dict.items():
    hash_vals[frozenset(val)].append(key)

# 合并具有相同集合值的键
res = {'-'.join(keys): vals for (vals, keys) in hash_vals.items()}

# 打印结果
print("合并后的键: ", res)

方法二：使用 groupby() + join() + 字典推导式

此方法首先使用 groupby() 对具有相同值的元素进行分组，然后利用 join() 连接这些键，最后通过字典推导式完成整个过程。

# Python3 代码示例
from itertools import groupby

# 初始化字典
test_dict = {'gfg': {5, 4, 3}, 'is': {4, 3, 5}, 'best': {1, 4, 3}, 'for': {1, 3, 4}, 'geeks': {1, 2, 3}}

# 打印原始字典
print("原始字典: ", test_dict)

# 使用 groupby() 对元素进行分组
# 字典推导式提供了一种简洁的方式来完成分组
res = {'-'.join(val): key for key, val in groupby(sorted(test_dict, key=test_dict.get), key=lambda sub: test_dict[sub])}

# 打印结果
print("合并后的键: ", res)