因为需要将返回的结果都写进表格,并输出折线图。但因为有较多不同的分类,故而需要创建不同的sheet表,废话不多说,直接上代码:
import openpyxl, os
from openpyxl.chart import LineChart, Reference
class WriteData():
def __init__(self, title, x_title):
self.title = title
self.x_title = x_title
'''
func: 创建结果表
title: 表的名字
'''
def create_excel(self):
# 检查文件是否存在
if os.path.isfile(self.title):
# 存在,则先删除表
os.remove(self.title)
# 创建工作表
self.wb = openpyxl.Workbook() # 创建工作薄
self.wb.save(self.title)
self.wb.close()
'''
打开表,在什么位置创建第n张表,并命名
sheet_title: 表名
index: 在什么位置创建表
x_title: 表头
'''
def open_create_sheet(self, sheet_title, index):
self.wb = openpyxl.load_workbook(self.title)
if os.path.isfile(self.title):
with open(s