用python写一个atm机的代码

好的，下面是一个简单的 Python ATM 机程序示例，您可以参考一下：

# 定义账户信息
accounts = {
    '001': {'name': '张三', 'password': '123456', 'balance': 1000},
    '002': {'name': '李四', 'password': '123456', 'balance': 2000},
    '003': {'name': '王五', 'password': '123456', 'balance': 3000}
}

# 登录函数
def login():
    account_no = input('请输入账号：')
    password = input('请输入密码：')
    account = accounts.get(account_no)
    if account and account.get('password') == password:
        return account
    else:
        print('账号或密码错误')
        return None

# 查询余额函数
def query_balance(account):
    print(f"您的账户余额为：{account.get('balance')}元")

# 取款函数
def withdraw(account):
    amount = float(input('请输入取款金额：'))
    if amount <= account.get('balance'):
        account['balance'] -= amount
        print(f"取款成功，当前余额为{account.get('balance')}元")
    else:
        print('余额不足，取款失败')

# 存款函数
def deposit(account):
    amount = float(input('请输入存款金额：'))
    account['balance'] += amount
    print(f"存款成功，当前余额为{account.get('balance')}元")

# 退出函数
def quit():
    print('感谢您使用本ATM机，再见！')

# 主函数
def main():
    print('欢迎使用本ATM机！')
    account = login()
    if account:
        while True:
            print('请选择操作：')
            print('1. 查询余额')
            print('2. 取款')
            print('3. 存款')
            print('4. 退出')
            choice = input()
            if choice == '1':
                query_balance(account)
            elif choice == '2':
                withdraw(account)
            elif choice == '3':
                deposit(account)
            elif choice == '4':
                quit()
                break
            else:
                print('输入有误，请重新输入')

# 运行主函数
if __name__ == '__main__':
    main()

这个程序中，我们定义了一个账户信息字典 accounts，包含了三个账户的信息，包括账号、密码和余额。然后定义了几个操作函数，包括登录、查询余额、取款、存款和退出。在主函数中，我们可以根据用户的输入选择相应的操作。当用户选择退出时，程序结束运行。