要让程序自动将图表保存到文件中,代码为:
plt.savefig('squares_plot.png', bbox_inches='tight')
第一个实参指定要以什么样的文件名保存图表,这个文件将存储到scatter_squares.py所在的目录中。
第二个实参指定将图表多余的空白区域裁减掉。如果要保留图表周围多余的空白区域,可省略这个实参。
如果发现保存的图表空白,检查一下是不是写成了这样:
plt.show()
plt.savefig('squares_plot.png', bbox_inches='tight')
将plt.show()
删除就可以了。