Matplotlib：检查是否未定义显示

Question

有时我通过 ssh 运行我的脚本。 This answer告诉我设置

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

当我通过 ssh 运行脚本得到 undefined SCREEN 错误时。然而，当我在本地机器上运行脚本时，我无法以交互方式查看图表。

检查屏幕是否定义的条件是什么？我想做

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')