用Python字典功能调用函数，极大优化代码结构！-CSDN博客

本文链接：https://blog.csdn.net/glb_0627/article/details/140876411

在这里插入图片描述

在使用Python编写程序时，可能需要根据用户的不同选择执行不同的函数。传统的`if-elif`结构虽然可以实现这一需求，但代码易变得复杂且难以维护。

本文将演示如何通过Python字典来实现这一功能，从而简化代码结构并提高可维护性。

一、函数定义与优化

首先，我们定义三个主要函数：

def convert_to_lower(s):`    `"""将字符串转换为小写"""`    `return s.lower()``   ``def convert_to_upper(s):`    `"""将字符串转换为大写"""`    `return s.upper()``   ``def print_alphabet():`    `"""输出字母A到Z"""`    `for letter in range(65, 91):`        `print(chr(letter), end=" ")

二、创建字典

创建一个字典`function_dict`，将用户选择与对应函数关联，并详细解释字典的`get`方法：

function_dict = {`    `"1": convert_to_upper,`    `"2": convert_to_lower,`    `"3": print_alphabet``}

三、实现功能选择

根据用户的输入选择调用相应的函数，并增加输入验证和错误处理：

while True:`    `print("""`    `1. 将字母全部改为大写`    `2. 将字母全部改为小写`    `3. 输出字母A-Z`    `""")``   `    `choice = input("请输入要使用的功能: ")`    `if choice in function_dict:`        `if choice == "1" or choice == "2":`            `user_input = input("请输入要处理的字符串: ")`            `print(function_dict[choice](user_input))`        `else:`            `function_dict[choice]()`    `else:`        `print("输入有误，请重新输入！")`    `print()

四、完整代码示例

def convert_to_lower(s):`    `"""将字符串转换为小写"""`    `return s.lower()``   ``def convert_to_upper(s):`    `"""将字符串转换为大写"""`    `return s.upper()``   ``def print_alphabet():`    `"""输出字母A到Z"""`    `for letter in range(65, 91):`        `print(chr(letter), end=" ")``   ``function_dict = {`    `"1": convert_to_upper,`    `"2": convert_to_lower,`    `"3": print_alphabet``}``   ``while True:`    `print("""`    `1. 将字母全部改为大写`    `2. 将字母全部改为小写`    `3. 输出字母A-Z`    `""")``   `    `choice = input("请输入要使用的功能: ")`    `if choice in function_dict:`        `if choice == "1" or choice == "2":`            `user_input = input("请输入要处理的字符串: ")`            `print(function_dict[choice](user_input))`        `else:`            `function_dict[choice]()`    `else:`        `print("输入有误，请重新输入！")`    `print()

运行结果如下：

五、拓展应用

除了以上示例，Python字典调用函数还可以应用于实现命令模式设计模式、菜单驱动程序和命令处理等。下面是一个简单的案例，用于演示如何使用字典实现简单的菜单驱动程序。

def add(a, b):`    `"""两个数相加"""`    `return a + b``   ``def subtract(a, b):`    `"""两个数相减"""`    `return a - b``   ``def multiply(a, b):`    `"""两个数相乘"""`    `return a * b``   ``def divide(a, b):`    `"""两个数相除"""`    `if b != 0:`        `return a / b`    `else:`        `return "除数不能为零"``   ``operation_dict = {`    `"1": add,`    `"2": subtract,`    `"3": multiply,`    `"4": divide``}``   ``while True:`    `print("""`    `1. 加法`    `2. 减法`    `3. 乘法`    `4. 除法`    `""")``   `    `choice = input("请选择操作: ")`    `if choice in operation_dict:`        `num1 = float(input("请输入第一个数字: "))`        `num2 = float(input("请输入第二个数字: "))`        `result = operation_dict[choice](num1, num2)`        `print("结果是: ", result)`    `else:`        `print("无效选择，请重新输入！")`    `print()

通过以上案例，我们可以看到如何通过字典灵活地实现函数调用，从而构建一个简单的计算器程序。

六、总结

在实际应用中，使用字典来调用函数可以大大简化菜单驱动程序、命令处理等常见编程任务。我们通过具体的示例展示了这种方法的优势，并提供了多个案例来帮助读者更好地理解和应用。

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后给大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！
在这里插入图片描述