优化Python循环与条件逻辑：高效处理动态数据的方法

admin 百科 2025-12-12 13

本文旨在探讨如何优化python代码中处理动态数据的循环与条件逻辑，避免冗长的`if/elif`链。通过重构数据结构（如使用嵌套字典），可以直接通过键值访问数据，从而实现更具扩展性、可维护性和效率的代码，特别适用于处理用户账户、配置项等场景。

在软件开发中，我们经常需要处理结构化但又动态变化的数据。一个常见的场景是，根据用户输入（如账户号码）来查找并操作对应的数据。初学者往往倾向于使用一系列的if或elif语句来匹配不同的条件，但这会导致代码冗长、难以维护且扩展性差。本教程将以一个银行账户管理系统的简化示例，展示如何通过优化数据结构来解决这一问题。

初始问题：冗余的条件判断

考虑一个简单的银行余额查询程序。它存储了多个用户的账户信息，并根据输入的账户号码显示余额。最初的实现可能如下所示：

print("welcome to bank!\n")
bank = {
    "sam": [1234, 150],
    "bob": [1235, 250],
}

print("please select an action:\n")
print(
    "1. check balance \n2. withdraw money \n3. deposit money \n4. exit\n"
)
action = int(input("selected action:"))
accnuminput = int(input("please enter account number:"))

if action == 1:
    if accnuminput == (bank["sam"][0]):
        print(bank["sam"][1])
    # 如果有更多账户，就需要添加更多类似的if语句
    # if accnuminput == (bank["bob"][0]):
    #     print(bank["bob"][1])

登录后复制

这种方法的问题在于，每增加一个账户，就需要手动添加一个if或elif语句来处理。当账户数量庞大时，代码将变得难以管理，且容易出错。这违反了“不要重复自己”（DRY - Don't Repeat Yourself）的编程原则。

解决方案：优化数据结构实现动态访问

解决上述问题的关键在于选择合适的数据结构，使得我们可以通过一个动态的键（例如用户输入的账户号码）直接访问所需的数据，而不是通过一系列硬编码的条件判断。Python的字典（dict）是实现这一目标的理想选择。

立即学习“Python免费学习笔记（深入）”；

我们可以将银行的数据结构进行重构，使其更合理地表示用户、账户及其余额之间的关系。一个更优化的结构可以是嵌套字典：最外层以用户名为键，内层字典则以账户号码为键，对应账户余额。

bank = {
    "sam": {1234: 150, 1236: 800}, # Sam有两个账户
    "bob": {1235: 250},
}

print("welcome to bank!\n")
user = input("please enter your name:")

# 检查用户是否存在
if user not in bank:
    print("user not found")
    exit()

print(f"hi {user}, your account numbers are {sorted(bank[user].keys())}")

print("please select an action:\n")
print("1. check balance \n2. withdraw money \n3. deposit money \n4. exit\n")
action = int(input("selected action:"))
accnuminput = int(input("please enter account number:"))

if action == 1:
    # 直接通过用户名和账户号码访问余额
    # 这里需要进一步的错误处理，例如账户号码不存在的情况
    if accnuminput in bank[user]:
        print(
            f"{user}, the balance of your account {accnuminput} is {bank[user][accnuminput]}"
        )
    else:
        print(f"Account number {accnuminput} not found for user {user}.")

登录后复制