2018年泰迪杯实验py部分代码

对于数据的读取保存和求得某一个月的销售额的实验代码：

import xlrd
import xlwt

from datetime import datetime
from xlrd import xldate_as_tuple

#设置表格样式
def set_style(name,height,bold=False):
    style = xlwt.XFStyle()
    font = xlwt.Font()
    font.name = name
    font.bold = bold
    font.color_index = 4
    font.height = height
    style.font = font
    return style


file = 'D:\\py代码\\泰迪杯\\附件1.xlsx'
data = xlrd.open_workbook(file)
table = data.sheets()[0]
#行
nrows = table.nrows
#列
ncols = table.ncols
print(nrows, ncols)
#第3行第七列的值
#print(table.cell(2,6).value)


book = xlwt.Workbook()
sheet = book.add_sheet('test')
print('flush_row_data' in dir(sheet))


default = xlwt.easyxf('font: name Arial;') # define style out the loop will work

row0 = ['订单号', '设备ID', '应付金额', '实际金额', '商品', '支付时间', '地点', '状态', '提现']
#写入满足第七列地点为A的数据信息
#写第一行
for i in range(0,len(row0)):
    sheet.write(0,i,row0[i],set_style('Times New Roman',220,True))
j = 1
sum=0
onesum=0
for i in range(1, nrows):
      # print(table.cell(i,6))
      # print(table.row_values(i))

    if table.cell(i, 6).value == 'A':
        try:
            one = datetime.strptime('2017-01-01 00:00:00', '%Y-%m-%d %H:%M:%S')
            two = datetime.strptime('2017-02-01 00:00:00', '%Y-%m-%d %H:%M:%S')
            # x就是你那串数字的变量

            if(datetime(*xldate_as_tuple(table.cell(i, 5).value, 0))>one and datetime(*xldate_as_tuple(table.cell(i, 5).value, 0))<two):
                onesum=onesum+table.cell(i, 3).value

            #求销售量
            sum=sum+int(table.cell(i, 3).value)
            # print("成立")
            # print(table.cell(i, 6))
            for k in range(0, len(row0)):
                # print(table.cell(i, k).value)
                sheet.write(j, k, table.cell(i, k).value,default)
            j = j + 1

        except:
            print(i)
            raise
book.save('test.xls')

print("A销售地点的所有销售额：",sum)
print("A销售地点1月份的所有销售额：",onesum)