用python编写购物程序(1)

要求：

1. 启动程序后，让用户输入工资，然后打印商品列表
2. 允许用户根据商品编号购买商品
3. 用户选择商品后，检测余额是否充足，够就直接扣款，不够就提醒
4. 可随时推出，退出时打印以购买商品，购买商品数量及余额

代码：

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:James Tao


salary=int(input('请输入您的工资:'))
list_of_goods=[['iphone',5800],['Mac Pro',12000],['Starbuck',31],['Bicycle',800]]
print('支持购买的商品及价格:',list_of_goods)
balance=salary
goods_of_bought=[]
goods_of_categorical={}

judge=True
while balance>0 and judge:

    #打印出商品列表及编号
    for i in range(len(list_of_goods)):
        print('{goods}对应编号为:{n} 
'.format(goods=list_of_goods[i][0],n=i))

    number = int(input('请输入您要购买的商品编号:'))
    #计算余额
    balance=balance-int(list_of_goods[number][1])

    #判断余额是否为0
    if balance>0:

        #将购买的商品加入购物车
        goods_of_bought.append(list_of_goods[number][0])
        quit1=input('继续购买?(Y?N):')
        if quit1=='N':
            judge=False

    else:

        #若余额小于0,将上一次购买的商品金额去除
        balance = balance + int(list_of_goods[number][1])
        quit2=input('余额不足,是否退出?(Y/N):')
        if quit2=='Y':
            judge=False

#判断是否购买了商品
if goods_of_bought: #如果列表为空等于False

    #统计购买的商品种类
    goods_of_set=set(goods_of_bought)

    #统计购买的商品数量并输出
    for item in goods_of_set:
        goods_of_categorical[item]=goods_of_bought.count(item)
    print('您购买的商品及数量为:',goods_of_categorical)

else:
    print('您未购买任何商品')

print('余额为:',balance)