代码整洁之道:用字典替换冗长的if-else语句

最近在写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开发中写出更简洁，更优雅的代码。你平时在代码优化中有什么技巧吗？欢迎在评论区分享你的经验！