计算余额

以下是一个示例代码，用于计算余额：

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

    def deposit(self, amount):
        self.balance += amount

    def withdraw(self, amount):
        if self.balance >= amount:
            self.balance -= amount
        else:
            print("余额不足，无法取款")

    def get_balance(self):
        return self.balance


# 创建一个银行账户对象
account = BankAccount(1000)

# 存款500元
account.deposit(500)

# 取款200元
account.withdraw(200)

# 获取余额
balance = account.get_balance()
print("余额：", balance)

这个示例代码定义了一个名为BankAccount的类，该类包含了存款、取款和获取余额的方法。在类的初始化方法中，可以传入一个初始余额，如果没有传入，则默认为0。存款和取款方法会根据输入的金额更新余额，获取余额方法会返回当前余额。

在示例中，首先创建了一个初始余额为1000的银行账户对象。然后进行存款和取款操作，最后通过调用get_balance()方法获取最新的余额，并打印输出。