简单购物车python实现 （初学者学习判断和列表练习）

'''简单购物车实现
 1，用户输入余额后，输出商品
 2，允许用户根据商品编号购买商品
 3，用户选择商品，余额够就买，不够就提醒
 4，商品数量实时更新，当数量为零时删除此商品
 5，随时可退出，退出时打印购买的商品和余额

'''

goods = [
    ["Iphone",6000,5],
    ["Mac pro",12000,1],
    ["bike",800,10],
    ["computer",4000,10],
    ["coffee",30,30],
    ["Linux",120,20]
]
shopping_list = []
num_list =[]#记录购买商品数量
salary = input("input your salary:")
if salary.isdigit():
    salary = int(salary)
    while True:
        print("编号\t产品\t价格\t数量")
        for index,item in enumerate(goods):#enumerate()方法是把列表的索引和对象输出
            print("%s\t\t%s\t%s\t%s"%(index,item[0],item[1],item[2]))
        user_choice = input("选择需要购买的商品编号(按q退出）>>>")
        if user_choice.isdigit():
            user_choice = int(user_choice)
            if user_choice < len(goods) and user_choice >=0:
                if goods[user_choice][1] < salary:
                    shopping_list.append(goods[user_choice][:2])
                    goods[user_choice][2] -=1 #商品数量-1
                    salary -= goods[user_choice][1]
                    #\033[33;1m%s\033[0m  让特定字符串变色  33为黄色
                    print("Added \033[33;1m%s\033[0m into your shopping cart,your current balance is \033[31;1m%s\033[0m"%(goods[user_choice][:2],salary))
                    if goods[user_choice][2] ==0:
                        goods.pop(user_choice)#  数量为零时 删除此商品
                else:
                    print("your current balance is not enough to buy this product...")
            else :
                print("The product is not exist!!!")
        elif user_choice == "q":
            print("----------shopping list----------")
            for i in shopping_list:
                print (i)
            print("Your current balance :",salary)
            exit()
        else:
            print("invalid input...")