python绘制折线图并保存

1.下面这部分代码是绘图并保存的代码

import numpy as np
import matplotlib.pyplot as plt
x=range(10)
y=[3,1,2,9,6,2,1,8,10,8]
l=plt.plot(x,y,"b--*")
plt.xlabel("epoch")
plt.ylabel("avg acc")
plt.legend()
plt.savefig("acc.png")
plt.show()

运行结果:
python绘制折线图并保存_第1张图片

2.下面是把折线图保存到指定路径下的代码:

import os
def getAccLineChart(x,y,save_dir,chart_name):
    l=plt.plot(x,y,"b--*")
    plt.xlabel("epoch")
    plt.ylabel("avg acc")
    plt.legend()
    if os.path.exists(save_dir)==False:
        os.makedirs(save_dir)
    plt.savefig(os.path.join(save_dir,chart_name))

x=range(10)
y=[3,1,2,9,6,2,1,8,10,8]
getAccLineChart(x,y,"../graph","1.png")

你可能感兴趣的:(python,python,开发语言,matplotlib)