python如何实现图片绘制并保存

# 如绘制图1：绘图并保存，绘制真实标签的能耗曲线，线颜色为红色，预测结果线形为虚线，对比图为fig1（图片的数据没放进去，要画图的话要有真实标签original，预测结果pred的值）

from matplotlib import pyplot as plt# 导入库
plt.plot(original, color = 'red', label = 'Observed')# 绘制原始数据
plt.plot(pred, '-.', label = 'Predicted')# 绘制预测数据
plt.title('Forecast with Updates')# 绘制标题
plt.xlabel('Time')# x坐标轴
plt.ylabel('Power consumption')# y坐标轴
plt.legend()
# 指定图片存储路径
figure_save_path = "file_fig"# 指定文件夹
if not os.path.exists(figure_save_path):
os.makedirs(figure_save_path) # 如果不存在目录figure_save_path，则创建
plt.savefig(os.path.join(figure_save_path , 'fig1.png'))#第一个是指存储路径，第二个是图片名字
plt.show() # 图片展示