创建一个名为BankAccount的类,要求:
__init__,接受账户名和初始余额;deposit(amount)方法用于存款;withdraw(amount)方法用于取款(余额不足时提示);display()方法显示账户信息;请根据要求,补全Python代码:
class BankAccount:
def ① (self, account_name, initial_balance):
self.account_name = account_name
self.balance = ②
def deposit(self, amount):
if amount > 0:
③
print(f"成功存入{amount}元")
else:
print("存款金额必须大于0")
def withdraw(self, amount):
if amount > self.balance:
print("余额不足")
else:
④
print(f"成功取出{amount}元")
def display(self):
print(f"账户名:{self.account_name}")
print(f"当前余额:{ ⑤ }元")
account = BankAccount("张三", 1000)
account.deposit(500)
account.withdraw(200)
account.withdraw(2000) # 测试余额不足情况
account.display()