数据集
要求如下:
1.将4个数据集折线图画在一张图上,只要第一列
2.有几个数据集,没有第一行,不能手动删除,使用代码删除第一行
导入需要的库
import matplotlib.pyplot as plt
import pandas as pd
为了防止中文乱码,需要加入两行代码如下:
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
读取csv文件数据
cbamtrain_data=pd.read_csv('unet ++ cbamtrain_loss.csv')
train_data=pd.read_csv('unet++ train_loss.csv')
jiehe_data=pd.read_csv('unet++ 结合空洞train_loss.csv',engine='python')
benwen_data=pd.read_csv('本文.csv',engine='python')
设置每个折线的x,y值
对y值取数据集的第一列,x为序列数
y1=cbamtrain_data.values[:,0]
x1=cbamtrain_data.index
y2=train_data.values[:,0]
x2=train_data.index
y3=jiehe_data.values[:,0]
x3=jiehe_data.index
y4=benwen_data.values[:,0]
x4=benwen_data.index
设置图片大小,开始进行画图
设置每个折线的颜色,为背景加上网格
fig=plt.figure(figsize=(12,8))
plt.plot(x1,y1,'b',x2,y2,'r',x3,y3,'g',x4,y4,'k')
plt.grid()
对图进行加工
plt.xlabel('Epoch',fontsize=20)
plt.ylabel('Loss',fontsize=20)
plt.legend(["unet ++ cbamtrain_loss","unet++ train_loss","unet++ 结合空洞train_loss","本文"],fontsize=15)
一个简单的折线图就画好了
代码
import matplotlib.pyplot as plt
import pandas as pd
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
cbamtrain_data=pd.read_csv('unet ++ cbamtrain_loss.csv')
train_data=pd.read_csv('unet++ train_loss.csv')
jiehe_data=pd.read_csv('unet++ 结合空洞train_loss.csv',engine='python')
benwen_data=pd.read_csv('本文.csv',engine='python')
y1=cbamtrain_data.values[:,0]
x1=cbamtrain_data.index
y2=train_data.values[:,0]
x2=train_data.index
y3=jiehe_data.values[:,0]
x3=jiehe_data.index
y4=benwen_data.values[:,0]
x4=benwen_data.index
fig=plt.figure(figsize=(12,8))
plt.plot(x1,y1,'b',x2,y2,'r',x3,y3,'g',x4,y4,'k')
plt.grid()
plt.xlabel('Epoch',fontsize=20)
plt.ylabel('Loss',fontsize=20)
plt.legend(["unet ++ cbamtrain_loss","unet++ train_loss","unet++ 结合空洞train_loss","本文"],fontsize=15)
plt.show()