Python 输出 元素

编写一个程序，根据控制台输入的所有事务日志计算银行帐户的净金额（使用while 循环，直到用户输入“END”才退出事务日志输入，并输出最终银行账户的净金额）。事务日志格式中D表示存款，而W表示提款

num = 0
while True:
    a = input("请输入:\n")
    b = a.split(' ')  # 输入字符串变为列表
    if b[0] == 'D':
        num += int(b[1])
    if b[0] == 'W':
        num -= int(b[1])
    if b[0] == 'END':
        print("您的账号余额为：%d" % num)
        break