python决策树画图

要在 Python 中绘制决策树图，可以使用 `graphviz` 库。

首先，你需要安装 `graphviz` 库及其相关软件，可以通过以下命令进行安装：

!pip install graphviz
!sudo apt-get install graphviz

然后，你可以使用 `sklearn.tree` 中的决策树模型来训练模型，例如：

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import export_graphviz
import graphviz

iris = load_iris()
X = iris.data
y = iris.target

clf = DecisionTreeClassifier()
clf.fit(X, y)

dot_data = export_graphviz(clf, out_file=None, 
                feature_names=iris.feature_names,  
                class_names=iris.target_names,  
                filled=True, rounded=True,  
                special_characters=True)  
graph = graphviz.Source(dot_data)  
graph

这样就可以绘制出决策树图了。你可以保存图像，也可以直接在 Jupyter Notebook 中显示。