问题描述:
利用plt.plot对多维数组进行画图时,使用plt.savefig保存的图片出现曲线叠加:
可以看出,这里有两组曲线。
图像示例代码
a=np.random.randint(0,10,size=[3,4,10])
b = np.linspace(0,1,num=10)
for i in range(a.shape[0]):
for j in range(a.shape[1]):
plt.plot(a[i][j], b, label='a{}'.format(j))
plt.xlabel('this is x')
plt.ylabel('this is y')
plt.title('this is a demo')
plt.legend(loc='lower left', fontsize='small')
plt.savefig('a{}.png'.format(i), bbox_inches='tight')
# plt.close() # 问题解决,添加plt.close
问题解决
在plt.savefig()之后添加plt.close(),将此次循环中的fig关闭,下次循环时打开新的fig,这样保存的图片就不是在原来的基础上再绘制的图片了。
其他
- 在测试例子的时候发现,如果在plt.savefig(),之后进行plt.show()的话,保存的图片也是正常的。因为show展示的图片你会叉掉,关闭了fig,和plt.close()一样的效果。
- plt.show()之后再plt.savefig()的话,保存的是全白图