如何将dtreeviz.trees生成的可视化图以PNG格式保存到本地硬盘

dtreeviz生成的可视化对象自带原生保存接口，不需要手动截图或者额外转码，直接调用对应方法即可导出和Notebook内渲染效果完全一致的PNG文件。

实现代码 #

在你原有代码的基础上，新增一行调用save()方法的语句即可：

from sklearn import tree
from dtreeviz.trees import dtreeviz # 用于树可视化
from matplotlib import pyplot as plt
import graphviz

ax2 = dtreeviz(
 RFC1.estimators_[0],
 X_train1, y_train1, target_name='Mask',
 scale=1.1, class_names=['0','1'])

# 原有语句：Jupyter内直接渲染展示图形
ax2

# 新增语句：保存为PNG格式到本地，括号内替换为你想要的存储路径
ax2.save("mask_decision_tree.png")

可选配置 #

• 如需导出高清图，可以在save()方法中传入scale参数调整分辨率，例如ax2.save("mask_decision_tree.png", scale=2)即可导出2倍分辨率的无模糊PNG文件。
• 如果运行保存时报错提示graphviz相关依赖缺失，请确认本地已经安装graphviz二进制程序（仅安装Python端的graphviz包不够，需要安装对应系统的原生graphviz程序）。

内容的提问来源于stack exchange，提问作者Karar Shah