如果在无桌面的命令行环境使用matplotlib.pyplot绘图会遇到如下的问题:
RuntimeError: Invalid DISPLAY variable
此时,可以通过添加以下命令避免:
import matplotlib.pyplot as plt
plt.switch_backend('agg')
然后利用plt.imshow绘图:
plt.figure()
plt.title('FitureName')
plt.imshow(img)
最后通过savefig函数存储图片以便查看:
plt.savefig('输出路径')
解决无桌面环境下matplotlib绘图问题
本文介绍在无桌面的命令行环境中使用matplotlib.pyplot进行绘图时,如何避免RuntimeError: Invalid DISPLAY variable错误。通过调整后端为'agg',可以成功绘图并保存图片。
287

被折叠的 条评论
为什么被折叠?



