provare questo?
import matplotlib,os
r = os.system('python -c "import matplotlib.pyplot as plt;plt.figure()"')
if r != 0:
matplotlib.use('Agg')
import matplotlib.pyplot as plt
fig = plt.figure()
fig.savefig('myfig.png')
else:
import matplotlib.pyplot as plt
fig = plt.figure()
plt.show()
Puoi rilevare direttamente se hai un display con il modulo OS in python.nel mio caso è
>>> import os
>>> os.environ["DISPLAY"]
':0.0'
Il codice seguente funziona per me in Linux e Windows (dove presuppone che ci sia un dispositivo di visualizzazione):
import os
import matplotlib
if os.name == 'posix' and "DISPLAY" not in os.environ:
matplotlib.use('Agg')
Vedi https://stackoverflow.com/a/1325587/896111.
Nota che la riga matplotlib.use('Agg')
deve apparire dopo la prima importazione di matplotlib
(altrimenti riceverai un errore).