如何让我的for循环对不同的参数执行相同的操作，而无需编写更多的案例？

Question

我编写了一个模拟银行余额查询器的代码，当你给它一个账号时，它会输出对应的余额。

print("欢迎来到银行！
")
bank = {
    "sam": [1234, 150],
    "bob": [1235, 250],
}

print("请选择一个操作：
")
print(
    "1. 查询余额 
2. 取款 
3. 存款 
4. 退出
"
)
action = int(input("选择的操作："))
accnuminput = int(input("请输入账号："))
if action == 1:
    if accnuminput == bank["sam"][0]:
        print(bank["sam"][1])

之前每增加一个账户时，我都采用添加一个新的elif来处理不同的情况，就像这样：

if accnuminput == bank["sam"][0]:
    print(bank["sam"][1])
if accnuminput == bank["bob"][0]:
    print(bank["bob"][1])

但随着账户数量增多，这种方式就不可行了。我想请教一下，有没有办法让同一个公式适用于任意查询的账号，而不必为每个新添加的账户都写一个新的条件分支？谢谢。

Abizern · Answer

将用户名存储到一个变量中（此处是从用户处获取输入），并将其作为 bank 字典中的键来使用。同时，我稍微调整了一下数据结构，以便单个用户可以拥有多个银行账户；现在（概念上）结构是 (用户名: {账号: 余额})。

如果用户输入了无效的账号，这段代码会因为KeyError而崩溃，不过这是下一步你需要解决的问题。

bank = {
    "sam": {1234: 150, 1236: 800},
    "bob": {1235: 250},
}

print("欢迎来到银行！
")
username = input("请输入您的姓名：")

if username not in bank:
    print("未找到该用户")
    exit()

print(f"你好，{username}，你的账号号码有：{sorted(bank[username])}")

print("请选择一个操作：
")
print("1. 查询余额 
2. 取款 
3. 存款 
4. 退出
")
action = int(input("选择的操作："))
accnuminput = int(input("请输入账号号码："))
if action == 1:
    print(f"{username}，你的账号{accnuminput}的余额是{bank[username][accnuminput]}")

下面是一个示例运行过程：

欢迎来到银行！

请输入您的姓名：sam
你好，sam，你的账号号码有：[1234, 1236]
请选择一个操作：
1. 查询余额 
2. 取款 
3. 存款 
4. 退出
选择的操作：1
请输入账号号码：1234
sam，你的账号1234的余额是150
```

Brian Campbell · Answer

替换原先的代码片段：

if accnuminput == bank["sam"][0]:

你可以检查银行账号“1234”是否存在于任意账户详情中（账户详情是指包含银行号码和余额的列表）。

所以最终的代码看起来会像这样：

print("欢迎来到银行！
")

bank = {
    "sam": [1234, 150],
    "bob": [1235, 250],
}

print("请选择一个操作：
")
print("1. 查询余额 
2. 取款 
3. 存款 
4. 退出
")

action = int(input("选择的操作："))
accnuminput = int(input("请输入账号号码："))

if action == 1:

    # 检查账号号码是否存在于银行字典的任何账户详情中
    if accnuminput in {account[0] for account in bank.values()}:

        # 使用输入的账号号码获取余额
        for name, acc_details in bank.items():
            if acc_details[0] == accnuminput:
                print(f"{name} 的余额是：{acc_details[1]}")
                break
        else:
            print("未找到该账户。")

但更为合适的解决方案是重构字典，使用银行账号作为键，名字和余额作为值：

print("欢迎来到银行！
")

bank = {
    1234: {"name": "sam", "balance": 150},
    1235: {"name": "bob", "balance": 250},
}

print("请选择一个操作：
")
print("1. 查询余额 
2. 取款 
3. 存款 
4. 退出
")

action = int(input("选择的操作："))
accnuminput = int(input("请输入账号号码："))

if action == 1:

    # 检查账号号码是否存在于银行字典中
    if accnuminput in bank:

        # 使用输入的账号号码获取余额
        account_details = bank[accnuminput]
        print(f"{account_details['name']} 的余额是：{account_details['balance']}")
    else:
        print("未找到该账户。")

这里我们使用的是嵌套字典，当然也可以使用字典的列表形式存储相同信息。