python——输出csv格式数据并通过matplolib进行绘图

将数据保存为csv文件格式

import csv
with open("data.csv", 'w') as csvfile:
       writer = csv.writer(csvfile)
       writer.writerow(['epoch', 'loss']) #一定要加中括号[]，不然每次都会

将 csv数据存入Python列表

import csv #导入csv模块，该模块包含于python标准库中

filename='data-1.csv'
with open(filename,'r')as file:
    #1.创建阅读器对象
    reader=csv.reader(file)
    #2.读取文件头信息
    header_row=next(reader)

    #3.保存第一列信息到列表
    epoches, losses=[],[]  #将最高气温、最低气温、日期储存在列表中
    for row in reader: #遍历reader列表
    
        epoch= row[0] 
        epoches.append(epoch) #将每行的第一个数据保存在epoches

        loss= row[1] #同上
        losses.append(loss) #将每行的第二个数据保存在losses
    print(epoches)
    print(losses)

结果信息：

[‘0’, ‘1’]
[‘104.10205623559784’, ‘11.862516771283065’]

将csv列表数据利用折线图可视化

import csv #导入csv模块，该模块包含于python标准库中
from matplotlib import pyplot as plt #从matplotlib中导入pyplot并重命名为plt
   
    #绘制气温图表
    fig=plt.figure(dpi=128,figsize=(8,6))#添加绘图窗口，可绘制多条曲线
    plt.plot(epoches,losses,c='red',alpha=0.6)#plot()函数，第一个参数x值，第二个y值，第三个图形颜色

    #设置图形的格式
    plt.title("Train_Model",fontsize=24)#图形标题
    plt.xlabel("epoch",fontsize=14)#x轴标题及字号
    fig.autofmt_xdate()#调用fig.autofmt_xdate()绘制斜的日期标签，以防日期彼此重叠
    plt.ylabel("loss",fontsize=14)
    plt.tick_params(axis='both',which='major',labelsize=8)#坐标轴格式

    #给图表区域着色
    plt.fill_between(epoches,losses, facecolor='red',alpha=0.6)
    """
    facecolor为填充区域颜色
    alpha为填充颜色的透明度，0表示完全透明，1表示完全不透明
    """

    #显示图表
    plt.show()

完整代码

import csv #导入csv模块，该模块包含于python标准库中
from matplotlib import pyplot as plt #从matplotlib中导入pyplot并重命名为plt
   
filename='data-1.csv'
with open(filename,'r')as file:
    #1.创建阅读器对象
    reader=csv.reader(file)
    #2.读取文件头信息
    header_row=next(reader)

    #3.保存第一列信息到列表
    epoches, losses=[],[]  #将最高气温、最低气温、日期储存在列表中
    for row in reader: #遍历reader列表
    
        epoch= row[0] 
        epoches.append(epoch) #将每行的第一个数据保存在epoches

        loss= row[1] #同上
        losses.append(loss) #将每行的第二个数据保存在losses
    print(epoches)
    print(losses)
    #——————————————————————绘制图表——————————————————————————————
    #绘制气温图表
    fig=plt.figure(dpi=128,figsize=(8,6))#添加绘图窗口，可绘制多条曲线
    plt.plot(epoches,losses,c='red',alpha=0.6)#plot()函数，第一个参数x值，第二个y值，第三个图形颜色

    #设置图形的格式
    plt.title("Train_Model",fontsize=24)#图形标题
    plt.xlabel("epoch",fontsize=14)#x轴标题及字号
    fig.autofmt_xdate()#调用fig.autofmt_xdate()绘制斜的日期标签，以防日期彼此重叠
    plt.ylabel("loss",fontsize=14)
    plt.tick_params(axis='both',which='major',labelsize=8)#坐标轴格式

    #给图表区域着色
    plt.fill_between(epoches,losses, facecolor='red',alpha=0.6)
    """
    facecolor为填充区域颜色
    alpha为填充颜色的透明度，0表示完全透明，1表示完全不透明
    """

    #显示图表
    plt.show()

参考博客：

• Python-CSV文件存储
• Python使用matplotlib数据可视化显示CSV文件数据