1.方法一：xlwt
1.1 安装包
pip install xlwt
1.2 保存数据到 Excel
import xlwtimport numpy as npimport random# 新建表格book = xlwt.Workbook(encoding='utf-8', style_compression=0)sheet = book.add_sheet('mysheet', cell_overwrite_ok=True) #设置多个sheet# 往第一行写入名称sheet.write(0, 0, "espisode")sheet.write(0, 1, "reward")episode=100reward=random.sample(range(50, 150), episode)for i in range(100):# 将数据写入前两列 sheet.write(i+1, 1, reward[i]) # 第i+1行0列sheet.write(i+1, 0, i+1) # 第i+1行1列# 保存表格book.save('data.xls')
效果：
2. 方法二：openpyxl
pip install openpyxl
from openpyxl import Workbook#新建对象wb=Workbook()#激活sheetsheet=wb.active#给sheet一个名字sheet.title='mysheet'# 增加一行的语法是# sheet.append(一维的list或tuple)sheet.append(['编号','姓名','年龄','性别'])sheet.append([1,'556',20,'男'])sheet.append([2,'770',22,'男'])# 保存wb.save('test.xlsx')
效果：
2.1超详细例子
自己代码例子：数据过大保存到第三四列
def save(train_episode, test_episode, train_reward, evaluate_reward):book = xlwt.Workbook(encoding='utf-8', style_compression=0)sheet = book.add_sheet('train', cell_overwrite_ok=True) # 设置多个sheetsheet2 = book.add_sheet('evaluate', cell_overwrite_ok=True)# 往第一行写入名称sheet.write(0, 0, "train_episode")sheet.write(0, 1, "train_reward")for i in train_episode:# 将数据写入前两列sheet.write(i+1, 1, train_reward[i]) # 第i+1行0列sheet.write(i+1, 0, i+1) # 第i+1行1列if i >=60000:sheet.write(0, 2, "train_episode")sheet.write(0, 3, "train_reward")for j in (train_episode-60000):sheet.write(j+1, 3, train_reward[i]) # 第i+1行0列sheet.write(j+1, 2, train_episode) # 第i+1行1列sheet2.write(0, 0, "episode")sheet2.write(0, 1, "evaluate_reward")for k in test_episode:# 将数据写入前两列sheet.write(k+1, 1, evaluate_reward[k]) # 第i+1行0列sheet.write(k+1, 0, k+1) # 第i+1行1列# 保存表格book.save('reward.xls')
3.xlrd读取excel文件并画图
3.1 安装以及相关报错
pip install xlrd
值得注意的是：文件格式要保存为xls【excel数据存储另存为xls比较稳妥】，直接改后缀名可能还会报错，报错如下：
Excel xlsx file; not supported
Unsupported format, or corrupt file: Expected BOF record; found b'<?xml ve'
3.2 代码例子：
import xlwtimport randomimport matplotlib.pyplot as pltimport xlrdplt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号x_data=[]y_data=[]data = xlrd.open_workbook('text.xls')table = data.sheets()[0]cap = table.col_values(1) #读取第二列数据cap1 = table.col_values(0)#读取第一列数据#print(cap) #打印出来检验是否正确读取for i in range(0,998):y_data.append(cap[i])x_data.append(cap1[i]*50) #对第一列数据扩大50倍plt.plot(x_data, y_data,color="#006bac")plt.title('标题')plt.legend()#标签plt.xlabel('episodes')plt.ylabel('Average reward')plt.show()
3.3 代码多图例子
import xlrdimport matplotlib.pyplot as plt#调节字体plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号#导入excel文件，以及第几张表data = xlrd.open_workbook('GDP2(已自动还原).xls')table = data.sheets()[0]#第一个图的数据t1 = table.col_values(1)tt = t1[1:90]xAxis1 = range(1929,)#第二个图的数据t2 = table.col_values(2)tu = t2[27:90]xAxis2 = range(1955,)#第三个图的数据t3 = table.col_values(3)tv = t3[20:90]xAxis3 = range(1948,)#第四个图的数据t4 = table.col_values(4)tw = t4[42:90]xAxis4 = range(1970,)#作图plt.plot(xAxis1, tt, label='美国GDP')plt.plot(xAxis2, tu, label='日本GDP')plt.plot(xAxis3, tv, label='英国GDP')plt.plot(xAxis4, tw, label='韩国GDP')plt.xlabel('年份')plt.ylabel('GDP现价')plt.title("各国GDP的变化")plt.legend()plt.show()
