Python读取csv文件并绘制曲线

前言

有时候我们的数据保存在csv文件中，但是想要更加直观的看出数据的好坏，最好利用matplotlib来画出曲线图

数据准备

我的数据格式如下：

在画图时，我需要把第一行去掉

# 去除第一个元素
xdata = xdata.drop(xdata.index[0])
ydata = ydata.drop(ydata.index[0])

画图

import matplotlib.pyplot as plt
import pandas as pd
from pylab import *                                 #支持中文
 
plt.rcParams['xtick.direction'] = 'in'  # 将x周的刻度线方向设置向内
plt.rcParams['ytick.direction'] = 'in'  # 将y轴的刻度方向设置向内
plt.rcParams['font.family'] = ['sans-serif']
plt.rcParams['font.sans-serif'] = ['SimHei']

# 读取数据
data = pd.read_csv('./data.csv') 
xdata = data.loc[:, 'tag']
ydata = data.loc[:, 'All']
# 去除第一个元素
xdata = xdata.drop(xdata.index[0])
ydata = ydata.drop(ydata.index[0])
# 创建一个10英寸宽，5英寸高的图形
plt.figure(figsize=(10, 5))
# 画图
plt.plot(xdata, ydata, color='blue',  linewidth=1.2, mec='r', mfc='w', label=u'curve')  # color可自定义折线颜色，marker可自定义点形状，label为折线标注
plt.legend()
plt.xlabel(u'训练轮数', size=12)
plt.ylabel(u'准确率', size=12)
# 保存图片
plt.savefig('data_rate.png') 
plt.show()

效果如下：