如何简化Python多条件if判断？以猜水果程序为例

我要写一个猜水果程序：从水果列表随机选一个不展示给用户，通过颜色等提示引导用户猜答案。最初用多条件if的写法有逻辑错误，后来写出了能运行的版本，但水果列表扩大时，重复写多个or会很冗余，想找更高效简洁的实现方式。

问题代码分析 #

错误版本（逻辑错误） #

import random
fruit = ["apple","banana","cherry","durian","lemon","orange","pineapple","watermelon"]
max = len(fruit)
rng = (random.randint(0,max-1))
correct = (fruit[rng])
if correct == "apple" or "cherry":
    print("red")
elif correct == "banana" or "lemon" or "pineapple":
    print("yellow")

print(fruit[rng])

这段代码的问题出在条件判断：if correct == "apple" or "cherry" 实际会被解析为 (correct == "apple") or "cherry"，而非空字符串"cherry"在布尔判断中是True，所以不管correct是什么，这个条件都会成立，导致逻辑完全错误。

可运行但冗余版本 #

import random
fruit = ["apple","banana","cherry","durian","lemon","orange","pineapple","watermelon"]
max = len(fruit)
rng = (random.randint(0,max-1))
correct = (fruit[rng])
if correct == "apple" or correct == "cherry":
    print("red")
elif correct == "banana" or correct == "lemon" or correct == "pineapple":
    print("yellow")

print(fruit[rng])

这段代码逻辑正确，但每加一种新水果或新颜色分类，都要重复写correct == XXX，扩展性极差。

优化实现方法 #

用字典映射来关联颜色和对应的水果列表，同时用random.choice()直接随机选水果，简化代码：

import random

# 建立颜色-水果的映射字典，扩展性强，新增水果/颜色直接修改字典即可
color_fruit_map = {
    "red": ["apple", "cherry"],
    "yellow": ["banana", "lemon", "pineapple"],
    # 可以继续添加其他颜色和水果，比如：
    # "green": ["watermelon"],
    # "orange": ["orange"]
}

# 把所有水果合并成一个列表，方便随机选取
all_fruits = [fruit for fruits in color_fruit_map.values() for fruit in fruits]
correct_fruit = random.choice(all_fruits)

# 根据选中的水果找对应的颜色提示
for color, fruits in color_fruit_map.items():
    if correct_fruit in fruits:
        print(f"提示：{color}")
        break

# 这里只是验证用，实际程序中不展示正确答案
print(f"正确水果：{correct_fruit}")

优化点说明 #

• 字典映射：把颜色和对应水果绑定，新增分类或水果时只需修改字典，不用改动条件判断逻辑，代码更易维护。
• 简化随机选取：用random.choice()直接从列表选元素，省去手动计算索引的步骤，代码更简洁。
• 通用判断逻辑：通过遍历字典找到对应颜色，不管有多少种颜色分类，逻辑都通用。

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