Matplotlib 绘图并保存图像

来源：互联网发布：淘宝登陆异常编辑：程序博客网时间：2024/05/16 06:27

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

1、导入文件：

import matplotlibmatplotlib.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))

接下来绘图保存：

plt.plot(train_X, train_Y, 'ro', label='Original data')plt.plot(train_X, sess.run(W) * train_X + sess.run(b), label='Fitted line')plt.savefig("examples.jpg")

但是执行plt.savefig时遇到段错误，将

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

该句删除后，成功绘图；

but，我还不知道why，希望后续可以理解。。。

0 0