决策树画图python

你可以使用Python中的Graphviz库来绘制决策树。以下是一个简单的例子：

首先，安装Graphviz库：

!pip install graphviz

然后，导入需要的库：

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

接下来，加载iris数据集，并训练一个决策树模型：

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

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

最后，使用export_graphviz函数将决策树导出为Graphviz格式，然后使用Source函数将其可视化：

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 = Source(dot_data)  
graph

这将在Jupyter Notebook中显示决策树图。如果你想将其保存为PDF或PNG文件，可以使用render方法：

graph.render("iris_decision_tree")