错误RuntimeError: Invalid DISPLAY variable

原因：matplotlib的backend中的FltkAgg, GTK, GTKAgg, GTKCairo, TkAgg , Wx or WxAgg这几个backend都要求有GUI图形界面的

首先查看了一下本机matplotlib的默认backend为：

user@home:/opt/user$ python
Python 3.6.3 |Anaconda custom (64-bit)| (default, Oct 13 2017, 12:02:49)
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import matplotlib.pyplot as plt
>>> plt.get_backend()
'Qt5Agg'

但是因为我上面没有GUI，所以会报错：RuntimeError: Invalid DISPLAY variable

为了解决这个问题，在代码中添加上：

plt.switch_backend('agg')

解决方法：指定不需要GUI的backend（Agg, Cairo, PS, PDF or SVG）

然后问题就解决了