简单matpoltlib绘图实践

初学者matplotlib折线图

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
plt.style.use('ggplot')
data1 = pd.read_csv('tip1.csv')#引用csv文件数据作为折线图数据
data1 = np.array(data1)#放在数组内方便切片使用
data2 = pd.read_csv('tip2.csv')
data2 = np.array(data2)
data3 = pd.read_csv('tip3.csv')
data3 = np.array(data3)
data4 = pd.read_csv('tip4.csv')
data4 = np.array(data4)
fig = plt.figure()#创建一个基础图
fig.patch.set_facecolor('blue')#底色为蓝色
fig.patch.set_alpha(0.6)#设置透明度
ax1 = fig.add_subplot(1,1,1)#向基础图中添加子图
ax1.patch.set_facecolor("orange")#子图设为橘色
ax1.patch.set_alpha(1.0)
ax1.plot(data1[:,0],'+',color='blue',linestyle='-',label="DCECA-Unet++",linewidth=2)#切取第一列数据
ax1.plot(data2[:,0],'+',color='red',linestyle='-',label="DCSE-Unet++",linewidth=2)
ax1.plot(data3[:,0],'+',color='green',linestyle='-',label="Unet",linewidth=2)
ax1.plot(data4[:,0],'+',color='purple',linestyle='-',label="Unet++",linewidth=2)
ax1.xaxis.set_ticks_position('bottom')#设置刻度线位置在x轴底部和y轴左侧
ax1.yaxis.set_ticks_position('left')
plt.xlabel("Epoch")#设置横纵标题
plt.ylabel("Loss")
plt.legend(loc='upper right')#将小标放在右上角
plt.savefig("sample.png",dpi=400,bbox_inches='tight')#保存图片并使图片变得更清晰
plt.show()

下图为代码展示: