Matplotlib: проверьте, не определено ли DISPLAY

Question

Иногда я запускаю свой скрипт через ssh. This answer сказал мне, чтобы установить

import matplotlib 
#matplotlib.use('Agg') # Must be before importing matplotlib.pyplot or pylab! 
import matplotlib.pyplot as plt 

, когда я получаю ошибку undefined SCREEN, запустив скрипт через SSH. Однако с этой преамбулой я не могу просматривать графики в интерактивном режиме, когда запускаю скрипт на своей локальной машине.

В каком состоянии проверить, установлен ли экран? Я бы хотел сделать

if SCREEN == None: 
    matplotlib.use('Agg') 

Каков правильный код для этого, как я могу это проверить?

Answer 1

Это выглядит как самый простой способ сделать это, чтобы проверить «DISPLAY» переменную окружения

import os 

# 'DISPLAY' will be something like this ':0' 
# on your local machine, and None otherwise 
if os.environ.get('DISPLAY') is None: 
    matplotlib.use('Agg')