老师你好,请问我这个代码除了啥问题,上一节课的内容也是这个问题,搞不出来了。。。
import xlrd
from xlutils.copy import copy
def get_data():
wb = xlrd.open_workbook('./02办公自动化/excel_file/data01.xlsx')
sh = wb.sheet_by_index(0)
'''
{
a:[{},{},{}],
b:[{},{},{}],
c:[{},{},{}]
}
'''
all_data = {}
for r in range(sh.nrows):
d = {"type":sh.cell_value(r,1),"name":sh.cell_value(r,2),"count":sh.cell_value(r,3),"price":sh.cell_value(r,4)}
key = sh.cell_value(r,0)
if all_data.get(key):
all_data[key].append(d)
else:
all_data[key] = [d]
return all_data
def save(data):
wb = xlrd.open_workbook('./02办公自动化/excel_file/data01.xlsx')
wb2 = copy(wb)
for key in data.key():
temp_sheet = wb2.add_sheet(key)
for i,d in enumerate(data.get(key)):
temp_sheet.write(i,0,d.get("key"))
temp_sheet.write(i,0,d.get("name"))
temp_sheet.write(i,0,d.get("count"))
temp_sheet.write(i,0,d.get("price"))
wb2.save('./excel_file/表格的拆分.xlsx')
if __name__ == "__main__":
all_data = get_data()
save(all_data)
