python求钱包余额代码

在Python中，我们可以创建一个简单的函数来模拟钱包余额的功能。假设我们有一个基础的`Wallet`类，它有一个初始余额和增加或减少余额的方法。这是一个例子：

class Wallet:
    def __init__(self, initial_balance=0):
        self.balance = initial_balance

    # 增加余额的方法
    def deposit(self, amount):
        if amount > 0:
            self.balance += amount
            return f"存款成功，当前余额为{self.balance}"
        else:
            return "请输入正数作为金额"

    # 减少余额的方法，如果余额足够
    def withdraw(self, amount):
        if amount <= self.balance:
            self.balance -= amount
            return f"取款成功，当前余额为{self.balance}"
        else:
            return "余额不足，无法完成取款"

# 使用示例
wallet = Wallet()
print(wallet.deposit(500))  # 输出: 存款成功，当前余额为500
print(wallet.withdraw(200))   # 输出: 取款成功，当前余额为300