论文实验绘图技巧（Python）

Pandas与Matplotlib结合，实现论文实验绘图

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-3,3,100)
y1 = 2*x + 1
y2 = x**2

#坐标轴范围
plt.xlim((-1,2))
plt.ylim((-2,3))

#横纵坐标描述
plt.xlabel('I AM X')
plt.ylabel('I AM Y')

l1, = plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
l2, = plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')

#图例
plt.legend(handles=[l1,l2],labels=['test1','test2'],loc='best')

plt.show()

x = np.linspace(0,200,1000)
y1 = 200 - 2*x
y2 = x**0.5

#xy范围
plt.xlim((0,200))
plt.ylim((0,200))

#设置中文
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus']=False

#xy描述
plt.xlabel('情节数')
plt.ylabel('迭代次数')

l1, = plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
l2, = plt.plot(x,y2,color='blue',linewidth=1.0,linestyle='-')

#画图例
plt.legend(handles=[l1,l2],labels=['IRL','GAN-IRL'])#,loc='right'


plt.show()

import pandas as pd

#读取csv文件
df = pd.read_csv('./cnn.csv')

#获取行列信息，设置字体大小7.5，图像大小（4,3）
df.plot(x='iter',y='accuracy', fontsize=7.5, figsize=(4,3))

#设置xy坐标范围
plt.xlim((0,1000))
plt.ylim((0.,1.))

#xy描述
plt.ylabel('准确率')
plt.xlabel('迭代次数')

#设置中文
plt.rcParams['font.sans-serif']=['SimHei']
plt.rcParams['axes.unicode_minus']=False

#刻度显示为in、out、inout
plt.rcParams['xtick.direction'] = 'in'
plt.rcParams['ytick.direction'] = 'in'

plt.savefig('./img_plot.png', dpi=600)# 设置dpi并保存图像到本地

plt.show()