最近在写Python代码的时候，你是否也遇到过这样的情况：为了处理多种情况，写了大量的if-else语句，代码不仅看起来冗长，而且维护起来也很麻烦？今天，我们就来聊聊如何使用Python中的字典来优化这种情况，让你的代码更加简洁高效。
开篇：if-else的困境
在日常编程中，我们经常需要根据不同的条件执行不同的操作。最常见的做法就是使用if-else语句。例如，我们要根据不同的食物名称返回不同的价格，可能会这样写：
def getPrice(food_item):
if food_item == "Burger":
return 100
elif food_item == "Pizza":
return 200
elif food_item == "Juice":
return 50
elif food_item == "Tofu":
return 150
这段代码虽然可以正常运行，但是当食物种类增加时，if-else语句会变得越来越长，代码的可读性和可维护性也会随之降低。这样的代码不仅看起来很乱，而且修改起来也很容易出错，给未来的代码维护留下了隐患。
核心：字典的优势
有没有更好的方法来解决这个问题呢？答案是肯定的，我们可以使用Python中的字典。字典是一种以键值对形式存储数据的数据结构，非常适合用来表示这种“条件-结果”的映射关系。我们可以将食物名称作为键，价格作为值，然后直接通过键来获取对应的值，而不需要写一堆if-else语句。
下面是用字典优化后的代码：
food_items_dict = {
"Burger": 100,
"Pizza": 200,
"Juice": 50,
"Tofu": 150
}
def getPrice(food_item):
return food_items_dict.get(food_item, 0)
这段代码简洁了很多，可读性也大大提高。我们使用了字典的 get() 方法来获取食物的价格。get() 方法的好处在于，如果字典中不存在对应的键，它会返回我们指定的默认值 (这里是 0)，而不会抛出异常。这样既避免了错误，也增加了代码的健壮性。
代码实践
我们来实际运行一下这两段代码：
# 第一种方法 (if-else)
def getPrice_if_else(food_item):
if food_item == "Burger":
return 100
elif food_item == "Pizza":
return 200
elif food_item == "Juice":
return 50
elif food_item == "Tofu":
return 150
# 第二种方法 (字典)
food_items_dict = {
"Burger": 100,
"Pizza": 200,
"Juice": 50,
"Tofu": 150
}
def getPrice_dict(food_item):
return food_items_dict.get(food_item, 0)
# 测试
food = "Pizza"
print(f"Price of {food} (if-else): {getPrice_if_else(food)}")
print(f"Price of {food} (字典): {getPrice_dict(food)}")
food = "Sushi"
print(f"Price of {food} (if-else): {getPrice_if_else(food)}") # 这里会报错
print(f"Price of {food} (字典): {getPrice_dict(food)}")
可以看到，使用字典不仅代码简洁，而且还更加健壮。当输入的食物不在字典中时，if-else 的方法会报错，而字典的方法会返回默认值0，防止程序崩溃。
进阶：match-case语句
当然，除了字典，在Python 3.10及以上版本中，我们还可以使用 match-case 语句来处理这种情况。match-case语句类似于其他语言中的 switch-case 语句，可以让代码更加清晰易懂。
def getPrice_match(food_item):
match food_item:
case "Burger":
return 100
case "Pizza":
return 200
case "Juice":
return 50
case "Tofu":
return 150
case _:
return 0
这里我们使用了 match 和 case 关键字，结构更加清晰。 case _: 代表了默认情况，如果前面的 case 都没有匹配上，就会执行默认的 return 0。但是相比较字典， match-case 语句在数据量很大的情况下，效率会低于字典。
总结与建议
在Python中，使用字典替代冗长的if-else语句可以大大提高代码的可读性和可维护性。字典不仅让代码更加简洁，还更加健壮。match-case 语句作为可选方案，也值得我们学习和使用。
- 优先使用字典： 当你需要在多个选项中进行选择时，优先考虑使用字典来存储选项和结果的映射关系。
- 善用get()方法： 使用字典的 get() 方法可以避免键不存在时抛出异常，同时可以设置默认值。
- 了解match-case语句： 在Python 3.10+版本中，可以尝试使用 match-case 语句，让代码更加清晰易懂。
希望这篇文章能够帮助你在日常的Python开发中写出更简洁，更优雅的代码。你平时在代码优化中有什么技巧吗？欢迎在评论区分享你的经验！