要求:
1.将4个数据集折线图画在一张图上,只要第一列
2.有几个数据集,没有第一行,不能手动删除,使用代码删除第一行
大致思路:
过程和结果:
import matplotlib.pyplot as plot
import pandas as pd
#防乱码
plot.rcParams['font.sans-serif'] = ['SimHei']
plot.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=plot.figure(figsize=(12,8))
plot.plot(x1,y1,'b',x2,y2,'r',x3,y3,'g',x4,y4,'k')
plot.grid()
plot.xlabel('Epoch',fontsize=20)
plot.ylabel('Loss',fontsize=20)
plot.legend(["unet ++ cbamtrain_loss","unet++ train_loss","unet++ 结合空洞train_loss","本文"],fontsize=15)
plot.show()