使用matplotlib模块存图

无GUI时，使用pyplot函数报如下错误：

_tkinter.TclError: no display name and no $DISPLAY environment variable

• 此时需要重启python环境，修改backend的值，如下：

>>> import matplotlib
>>> matplotlib.use('Agg')
>>> import matplotlib.pyplot as plt
>>> plt.figure()
<Figure size 640x480 with 0 Axes>

替换显示函数为存图savefig函数

import numpy as np

x = np.linspace(0, 2, 100)

plt.plot(x, x**3, label='cubic')

plt.xlabel('x label')  
plt.ylabel('y label')
plt.title("Simple Plot")
plt.legend(loc="upper left")

plt.savefig('cubic.jpg')

参考文献：

1. what is backend?