一、解決警告信息
1 ... 2 _tkinter.TclError: no display name and no $DISPLAY environment variable
兩種解決方法:
1、python文件中:
在 from matplotlib import pylot 之前,添加代碼:
如下:
1 import matplotlib as mpl 2 mpl.use('Agg') 3 4 from matplotlib import pylot
2、在配置文件中:
Linux中,創建文件 ~/.config/matplotlib/matplotlibrc,
(其中,~/.config/matplotlib/是配置文件matplotlibrc的路徑)
添加如下一行:
1 backend : Agg
Windows下解決方案應該是一樣的(只是配置文件位置不同),還沒有使用Windows跑過,請自行Google。
二、生成圖片
支持的格式(注意不支持jpg格式):
eps, pdf, pgf, png, ps, raw, rgba, svg, svgz
在plt.draw() 或者 plt.show() 之后添加生成圖片的代碼:
1 ... 2 plt.show() 3 plt.savefig('/home/moxiao/code/python/classification_pr.png')