一、任务要求:
1.将4个数据集折线图画在一张图上,只要第一列 2.有几个数据集,没有第一行,不能手动删除,使用代码删除第一行
二、具体实现:
1、需要用到的库:
import pandas as pd
import matplotlib.pyplot as plt
在这插一个在pycharm中安装pandas库时。需要先更新一下pip,否则会安装失败。 在cmd中输入;
python -m pip install -U pip
2、采用的风格:
plt.style.use('ggplot')
3、防止中文出现错误:
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
4、读取文件:
data1 = pd.read_csv('unet ++ cbamtrain_loss.csv')
data2 = pd.read_csv('unet++ train_loss.csv')
data3 = pd.read_csv('unet++ 结合空洞train_loss.csv',engine='python')
data4 = pd.read_csv('本文.csv',engine='python')
5、读入第一列:
y1=data1.values[:,0]
x1=data1.index
y2=data2.values[:,0]
x2=data2.index
y3=data3.values[:,0]
x3=data3.index
y4=data4.values[:,0]
x4=data4.index
6、设置折线的颜色,坐标轴,图片大小
fig=plt.figure(figsize=(18,10))
plt.xlabel('Epoch',fontsize=28)
plt.ylabel('Loss',fontsize=28)
plt.plot(x1,y1,'y-.',x2,y2,'c-.',x3,y3,'r-.',x4,y4,'b-.')
plt.legend(["unet ++ cbamtrain_loss","unet++ train_loss","unet++ 结合空洞train_loss","本文"],fontsize=20)
三、汇总:
代码实现: 效果图:
|