Python：列表元素匹配字典值列表生成对应键列表的代码问题

让我来帮你梳理代码里的问题，然后给出符合需求的实现方案～

代码错误分析 #

你的代码主要有三个问题：

• 遍历逻辑颠倒：需求是处理输入列表里的每一个字符串，你却先遍历字典的键值对，完全搞反了处理顺序。
• 判断条件错误：if value in lis是在检查整个字典值列表（比如['man', 'men', 'mun']）是否是输入列表lis的元素，但lis里全是单个字符串，这个条件永远不会成立——这就是你几乎得不到匹配结果的核心原因。
• 未初始化输出列表：代码里直接调用output.append(key)，但没有提前定义output = []，运行时会触发NameError。

基础实现方案（适合小数据量） #

我们调整遍历顺序，先处理输入列表的每个元素，再去字典里找对应的键：

lis = ['man', 'men', 'boy', 'buoy', 'cat','caat']
dic = {'man':['man', 'men', 'mun'], 'boy':['boy','buoy','bay'], 'cat':['cat','caat','cut']}

output = []
# 遍历输入列表的每个字符串
for s in lis:
    # 遍历字典的键值对，查找匹配的key
    for key, values in dic.items():
        if s in values:
            output.append(key)
            # 找到匹配后跳出内层循环，避免重复匹配（若存在多个key对应同一字符串的情况）
            break

print(output)
# 输出结果：['man', 'man', 'boy', 'boy', 'cat', 'cat']

优化实现方案（适合大数据量） #

如果你的字典和输入列表都比较大，双重循环的效率会很低。我们可以先构建一个反向映射字典（字符串→对应键），之后查找每个元素都是O(1)的时间复杂度：

lis = ['man', 'men', 'boy', 'buoy', 'cat','caat']
dic = {'man':['man', 'men', 'mun'], 'boy':['boy','buoy','bay'], 'cat':['cat','caat','cut']}

# 构建反向映射：将每个字符串映射到对应的字典键
reverse_map = {}
for key, values in dic.items():
    for val in values:
        reverse_map[val] = key

# 快速生成输出列表
output = [reverse_map[s] for s in lis]

print(output)
# 输出结果：['man', 'man', 'boy', 'boy', 'cat', 'cat']

这个优化方案只需要遍历字典一次构建映射，之后处理输入列表的速度会快很多，非常适合大规模数据场景。

内容的提问来源于stack exchange，提问作者sandeep gunasekar