解决matplotlib中的Invalid DISPLAY variable错误

在本地pyplot画图可以运行，但是在服务器显示以下错误：

RuntimeError: Invalid DISPLAY variable

其实这是因为matplotlib是默认画图backend是TkAgg，这个需要有GUI的图形界面。只需要指定不需要GUI的backend就可以解决这个问题：

1.

import matplotlib.pyplot as plt
plt.switch_backend('agg')

2.

如果上面不行的话，可以试试：

import matplotlib as mpl
mpl.use('Agg')

值得注意的是，这个必须要写在import pylab as plt之前。