如何用Python将一个字典根据 key 值拆分成多个字典？

def split_dict_by_keys(original_dict, key_groups):
    """
    根据指定的key分组将字典拆分为多个子字典
    
    Args:
        original_dict: 原始字典
        key_groups: 包含多个key列表的列表，每个列表代表一个子字典需要的keys
    
    Returns:
        包含所有子字典的列表
    """
    result = []
    for keys in key_groups:
        sub_dict = {k: original_dict[k] for k in keys if k in original_dict}
        result.append(sub_dict)
    return result

# 使用示例
data = {'a': 1, 'b': 2, 'c': 3, 'd': 4, 'e': 5}
groups = [['a', 'b'], ['c', 'd', 'e']]

split_dicts = split_dict_by_keys(data, groups)
print(split_dicts)  # 输出: [{'a': 1, 'b': 2}, {'c': 3, 'd': 4, 'e': 5}]

# 如果某些key不存在，会自动跳过
groups2 = [['a', 'x'], ['b', 'c']]  # 'x'不存在
split_dicts2 = split_dict_by_keys(data, groups2)
print(split_dicts2)  # 输出: [{'a': 1}, {'b': 2, 'c': 3}]

这个函数的核心是字典推导式，遍历每个key组，只保留原始字典中存在的key。如果你想要更灵活的方式，比如按条件动态分组，可以这样：

def split_dict_by_condition(original_dict, condition_func):
    """
    根据条件函数拆分字典
    
    Args:
        condition_func: 接收key和value的函数，返回分组标识
    """
    from collections import defaultdict
    
    result = defaultdict(dict)
    for key, value in original_dict.items():
        group_id = condition_func(key, value)
        result[group_id][key] = value
    
    return list(result.values())

# 示例：按key的类型分组
data = {'a': 1, 'b': 2, 'c1': 3, 'd2': 4}
split_by_type = split_dict_by_condition(data, lambda k, v: '字母' if k.isalpha() else '数字')
print(split_by_type)  # [{'a': 1, 'b': 2}, {'c1': 3, 'd2': 4}]

用字典推导式按key分组最直接。

from collections import defaultdict
ret = defaultdict(dict)
for k,v in a.items():
ret[k[:2]].update({k:v})