import matplotlib.pyplot as plt

import pandas as pd

from matplotlib.legend_handler import HandlerLine2D

record=pd.read_csv('file.csv') #pandas读取保存的csv文件

plt.figure(figsize=(15, 8)) #图片横纵比

plt.legend() #显示图例

l1, =plt.plot(range(3000),record['Episode Reward'],

color='g',ms=12, linewidth=2.5) #画线 3000是横坐标相当于x值 Episode Reward相当于y值

plt.xlim(0, 3000) #横坐标范围

plt.ylim(0, 3500) #纵坐标范围

plt.legend(

[l1], ['la_1'], #la_1是图例

handler_map={l1: HandlerLine2D(numpoints=1)},

loc=2, fontsize=15)

plt.grid() #显示网格

plt.xlabel('EP',fontsize=15) #横坐标标签以及字体大小

plt.ylabel('Episode Reward',fontsize=15)

plt.xticks(fontsize=15) #x轴刻度字体大小

plt.yticks(fontsize=15)

plt.savefig("save_file.pdf") #保存文件为pdf格式

plt.show() #显示图片

plt.plot() 中 color配色如图所示:

推荐文章

评论可见,请评论后查看内容,谢谢!!!
 您阅读本篇文章共花了: