如何在Python字典中拆分多值键，实现按顺序查询配对元素

你的字典使用二元元组作为键，需求是仅匹配元组第一个位置的元素，返回对应第二个元素的列表；若查询的是元组第二个位置的元素，则返回None。以下是两种实用实现方式：

1. 直接遍历查询（适合偶尔查询） #

先定义原始字典：

my_dict = { 
    ('X', 'A'): 5,
    ('X', 'B'): 2,
    ('Y', 'B'): 3,
    ('Z', 'A'): 3,
    ('Z', 'C'): 4,
}

编写函数实现查询逻辑：

def get_paired(target, data_dict):
    matched = []
    for key in data_dict:
        # 仅匹配二元元组的第一个元素
        if isinstance(key, tuple) and len(key) == 2 and key[0] == target:
            matched.append(key[1])
    # 有匹配项返回列表，否则返回None
    return matched if matched else None

测试示例：

• get_paired('X', my_dict) → 返回 ['A', 'B']
• get_paired('A', my_dict) → 返回 None
• get_paired('Y', my_dict) → 返回 ['B']

Python 3.7+的普通字典会保留插入顺序，所以返回的列表顺序和字典定义时的顺序一致。

2. 预转换字典（适合频繁查询） #

如果需要多次查询，预先将字典转换为以元组第一个元素为键的结构，能大幅提升查询效率：

# 预构建查询用的字典
indexed_dict = {}
for (first_elem, second_elem), _ in my_dict.items():
    if first_elem not in indexed_dict:
        indexed_dict[first_elem] = []
    indexed_dict[first_elem].append(second_elem)

查询时直接使用get方法，不存在的键自动返回None：

indexed_dict.get('X')  # 输出: ['A', 'B']
indexed_dict.get('A')  # 输出: None

这种方式的查询时间复杂度为O(1)，比每次遍历字典更高效。

内容的提问来源于stack exchange，提问作者06ov