如何基于用户输入为Python字典生成单值映射新字典

需求说明 #

我有如下Python字典：

dict_ = {
    'one': ['a','b','c','d'],
    'two': ['e','f','g'],
    'three': ['x']
}

需要针对值为多元素列表的键，提示用户从列表中选择一个值，最终生成每个键对应单个值的新字典。比如用户为one选择b、为two选择f时，期望得到：

new_dict = {'one': 'b', 'two':'f', 'three':'x'}

我原本的实现是先创建中间字典dupl_dict存储值为多元素列表的键值对，再遍历这个中间字典处理输入，但想知道能不能不创建这个中间字典来实现需求？

解决方案 #

完全可以不用中间字典，直接遍历原字典的同时构建新字典，一步到位：

dict_ = {
    'one': ['a','b','c','d'],
    'two': ['e','f','g'],
    'three': ['x']
}

new_dict = {}
for key, values in dict_.items():
    if len(values) > 1:
        # 增加简单输入验证，避免无效选择
        while True:
            user_choice = input(f"Choose value for {key} from {values}: ")
            if user_choice in values:
                break
            print(f"无效选项！请从{values}中选择")
        new_dict[key] = user_choice
    else:
        # 列表仅含单个元素，直接取值
        new_dict[key] = values[0]

print(new_dict)

代码说明 #

• 直接初始化空字典new_dict存储最终结果
• 遍历原字典的每一组键值对：
  • 若值的列表长度大于1，循环提示用户输入，直到输入内容在目标列表内（避免无效输入），再将选择的值存入new_dict
  • 若列表仅含单个元素，直接取出该元素存入new_dict
• 全程无需额外中间字典，一次遍历完成所有操作

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