matplotlib 小任务

数据集

要求如下：

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()

最终效果