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配色如图所示: