解決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之前。