Matplotlib 绘图并保存图像

下述代码源于参考TensorFlow入门一中线性回归的实现，想将结果图片保存查看。

1、导入文件：

import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

前两行是我添加的，因为我使用的是tensorflow1.0.0版本，运行.py文件中出现错误提示：

File "/usr/lib64/python2.7/lib-tk/Tkinter.py", line 1745, in __init__
    self.tk = _tkinter.create(screenName, baseName, className, interactive, wantobjects, useTk, sync, use)
_tkinter.TclError: no display name and no $DISPLAY environment variable

添加上这两行后错误消失；

2、准备绘图：

因为第一次使用python绘图，我先是写下以下语句想画个500*500的窗口：

plt.figure(figsize=(500, 500)) 

接下来