論文實驗繪圖技巧（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()