声明:博主为数学建模使用编写的转Excel代码,方便Excel绘图并调制;当然都转成列表数据了可以直接调用matplotlib库进行绘图,但是个人觉得转为Excel画图比较快、调整起来快一点;主要看个人习惯。
1、将列表数据写入Excel的代码模板
模板使用操作:直接将列表数据代入函数参数
1 import openpyxl 2 import xlwt 3 import xlsxwriter 4 import openpyxl as xl 5 import os 6 import numpy as np 7 import pandas as pd 8 9 10 # ---------------------------用xlwt库写入xls文件--------------------------- 11 def list_into_xls(data_list, excelname): 12 """ 13 用xlwt库写入Excel文档智能保存为xls文件,这种文件 是一个特有的二进制格式, 14 其核心结构是复合文档类型的结构,而 xlsx 的核心结构是 XML 类型的结构,采用的是基于 XML 的压缩方式,使其占用的空间更小。 15 :param data_list: 需要写入Excel的列表数据 16 :param excelname: 对Excel进行命名 17 :return: 返回一个写入的Excel文件,保存在当前代码文件所在的目录下 18 """ 19 book1 = xlwt.Workbook(encoding='utf-8') # 创建工作簿,相当于创建Excel文档 20 sheet1 = book1.add_sheet('sheet1', cell_overwrite_ok=True) # 添加名为sheet1的工作表 21 for i in range(len(data_list)): 22 # 注意用xlwt库写入时行列索引均从0开始 23 sheet1.write(i, 0, data_list[i]) # 默认将列表里的数据全部写到Excel里的第一列,方便数模画图 24 book1.save('{}.xls'.format(excelname)) 25 print('xlx格式表格写入数据成功!') 26 27 28 # ---------------------------用openpyxl库写入xlsx文件--------------------------- 29 def write_excel_xlsx(data_list, excelname): 30 # 函数参数同上,只不过返回的是xlsx类型的文档 31 index = len(data_list) 32 workbook = openpyxl.Workbook() 33 sheet = workbook.active 34 for i in range(index): 35 # 注意用openpyxl库写入时行列索引均从1开始 36 sheet.cell(row=i + 1, column=1, value=data_list[i]) 37 workbook.save('{}.xlsx'.format(excelname)) 38 print("xlsx格式表格写入数据成功!")
2、将dataframe类型数据写入Excel文档中
模板使用操作:直接将dataframe类型数据代入函数参数,然后调用函数即可
1 def dataframe_into_xlsx(data_frame, excelname): 2 """ 3 参数说明 4 :param data_frame:dataframe类型的数据 5 :param excelname: Excel命名的名字,随便命名,为一个字符型数据 6 :return: 返回生成在当前代码所在目录下的xlsx类型的文档 7 """ 8 data_frame.to_excel('{}.xlsx'.format(excelname), sheet_name='sheet1')