在for循环语句中,每循环一次就要生成一张图片,可以按照以下方法将照片自动保存:
1.
plt.savefig("D:/figures/temp{}.png".format(i))
2.
plt.clf()
示例:
#导入两个包
import matplotlib.pyplot as plt
import numpy as np
#数据准备
x = np.arange(27)
x = np.reshape(x, (3,9))
#建立for循环语句,绘制x的前三列
for i in range(3):
plt.plot(x[:,i])
# plt.show() #保存图片的时候不要plt.show()
plt.savefig("D:/figures/temp{}.png".format(i)) #输入地址,并利用format函数修改图片名称
plt.clf() #需要重新更新画布,否则会出现同一张画布上绘制多张图片
#我的代码记录
for l in hi_curve_all_engines:
N=20
ls = np.convolve(l, np.ones((N,))/N, mode='same')
lv = np.convolve(l, np.ones((N,))/N, mode='valid')#利用convolve函数对数据做滑动平均
lf = np.convolve(l, np.ones((N,))/N, mode='full')
plt.plot(l)
#plt.plot(ls)
plt.plot(lv) #选择一种滑动平均的方式
#plt.plot(lf)
plt.ylim((0, 1.2))
#plt.show()
i+=1
plt.savefig("D:/figures/HItrain{}.png".format(i))#保存图片
plt.clf()#更新画布