Databricks环境下决策树dot转png时报dot未找到错误如何解决

该报错核心原因是你的代码依赖的dot工具属于系统级graphviz软件包，仅安装Python侧的pydot、graphviz等第三方库无法解决问题，需按以下步骤操作：

1. 安装系统级graphviz包 #

你可以根据使用场景选择临时安装或永久安装：

• 临时安装（仅当前notebook会话生效，适合调试验证）：
  在notebook中新建单元格运行以下shell命令：

%sh
apt-get update
apt-get install -y graphviz

安装完成后可运行%sh which dot验证，输出/usr/bin/dot即代表安装成功。

• 永久安装（集群所有节点重启后依然生效，适合生产环境）：
  将上述apt命令写入.sh格式的初始化脚本，上传到DBFS路径后，在集群配置的「Init Scripts」板块添加对应脚本路径，重启集群即可完成全节点预装。

2. 手动添加PATH路径（可选，仅安装后仍报错时执行） #

如果完成安装后依然触发路径找不到的报错，可在你的业务代码前添加以下代码，手动将dot的安装路径加入Python运行时环境变量：

import os
os.environ["PATH"] += os.pathsep + "/usr/bin"

3. 运行原有业务代码 #

完成上述配置后，你的原有代码即可正常执行：

(graph,) = graph_from_dot_file('tree.dot')
graph.write_png('/mnt/tree1.png')

内容的提问来源于stack exchange，提问作者Pedro Paulo S. Argôlo