画训练折线图

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import scienceplots
plt.rcParams['font.sans-serif'] = 'simhei'
plt.rcParams['axes.unicode_minus'] = False

data = pd.read_csv("./raf_ce_curline.csv")
data = np.array(data)
print(data)

# data是一个多维数组，所以可以用data[:,1]这种分片操作取某一列的值
# plt.pie(data[:, 1], labels=data[:, 0], autopct="%.1ff%%")
# plt.legend(data[:, 0], loc="upper left")
# plt.show()
with plt.style.context(['science', 'grid']):
    plt.figure(figsize=(6, 6))

    r='#63b2ee'
    b='#f89588'
    y = '#f89588'
    a ='#eddd86'
    d ='#9987ce'
    e = '#9987ce'
    plt.plot(data[:,0],data[:,1], marker='o',markersize=2, color=r,markerfacecolor =b,markeredgecolor=y,label='Line 1')
    plt.plot(data[:,0],data[:,3], marker='o',markersize=2,color=a,markerfacecolor =d,markeredgecolor=e, label='Line 2')

    # 坐标轴名称
    plt.xlabel('epoch')
    plt.ylabel('ACC')

    # 图例
    plt.legend()

    plt.show()