如何优化Python中的嵌套if…elif…else？

Python是一门高级编程语言，常被用于开发各种应用程序，包括机器学习、数据分析等多个领域。在编写Python代码时，我们经常会遇到多个条件判断的情形，使用if…elif…else语句是一种常见的方式。但是，如果if…elif…else语句嵌套过多，会导致代码难以维护和理解。本文将介绍如何优化Python中的嵌套if…elif…else语句。

阅读更多：Python 教程

优化方法一：使用字典

在Python中，可以使用字典来代替if语句。以一个简单的判断用户角色的例子为例：

role = input("请输入用户角色：")

if role == "admin":
    print("欢迎管理员！")
elif role == "user":
    print("欢迎用户！")
else:
    print("您的角色未识别！")

可以使用字典来代替if语句，代码如下：

role_dict = {"admin": "欢迎管理员！", "user": "欢迎用户！"}

role = input("请输入用户角色：")
print(role_dict.get(role, "您的角色未识别！"))

在这个例子中，我们使用一个字典来存储不同用户角色的信息，使用get方法获取输入的角色对应的信息。如果输入的角色不在字典中，就输出“您的角色未识别！”。

使用字典可以避免if语句的嵌套，提高代码可读性。

优化方法二：使用函数

在Python中，可以定义函数来代替if语句。以一个判断学生成绩等级的例子为例：

score = int(input("请输入学生成绩："))

if score >= 90:
    print("优秀")
elif score >= 80:
    print("良好")
elif score >= 70:
    print("中等")
elif score >= 60:
    print("及格")
else:
    print("不及格")

可以使用函数来代替if语句，代码如下：

def get_grade(score):
    if score >= 90:
        return "优秀"
    elif score >= 80:
        return "良好"
    elif score >= 70:
        return "中等"
    elif score >= 60:
        return "及格"
    else:
        return "不及格"

score = int(input("请输入学生成绩："))
print(get_grade(score))

在这个例子中，我们定义了一个函数get_grade，根据输入的学生成绩返回对应的等级。使用函数可以避免if语句的嵌套，提高代码可读性。

优化方法三：使用多态

在Python中，可以使用多态来代替if语句。以一个简单的计算器程序为例：

operator = input("请输入计算符号：")
num1 = int(input("请输入第一个数字："))
num2 = int(input("请输入第二个数字："))

if operator == "+":
    print(num1 + num2)
elif operator == "-":
    print(num1 - num2)
elif operator == "*":
    print(num1 * num2)
elif operator == "/":
    print(num1 / num2)
else:
    print("计算符号输入错误！")

可以使用多态来代替if语句，代码如下：

class Calculator:
    def __init__(self, num1, num2):
        self.num1 = num1
        self.num2 = num2

    def calculate(self):
        pass

class Add(Calculator):
    def calculate(self):
        return self.num1 + self.num2

class Subtract(Calculator):
    def calculate(self):
        return self.num1 - self.num2

class Multiply(Calculator):
    def calculate(self):
        return self.num1 * self.num2

class Divide(Calculator):
    def calculate(self):
        try:
            return self.num1 / self.num2
        except ZeroDivisionError:
            print("除数不能为0！")

operator = input("请输入计算符号：")
num1 = int(input("请输入第一个数字："))
num2 = int(input("请输入第二个数字："))

if operator == "+":
    calc = Add(num1, num2)
elif operator == "-":
    calc = Subtract(num1, num2)
elif operator == "*":
    calc = Multiply(num1, num2)
elif operator == "/":
    calc = Divide(num1, num2)
else:
    print("计算符号输入错误！")
    exit()

result = calc.calculate()

if result is not None:
    print(result)

在这个例子中，我们定义了一个Calculator基类和四个子类Add、Subtract、Multiply、Divide，分别代表加减乘除四种操作。每个子类都实现了calculate方法，在实例化时根据输入的计算符号来选择对应的子类进行计算，避免了if语句的嵌套。