python决策树画图

要在Python中绘制决策树，通常需要使用Graphviz库。下面是一个简单的例子，展示如何使用DecisionTreeClassifier和Graphviz库绘制决策树。

首先，你需要安装Graphviz库。可以使用以下命令在终端中安装：

pip install graphviz

接下来，导入必要的库：

from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
from sklearn.externals.six import StringIO  
from IPython.display import Image  
import pydotplus

然后，加载一个示例数据集：

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

使用DecisionTreeClassifier拟合数据：

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

最后，使用export_graphviz函数将决策树导出为Graphviz格式，并使用pydotplus库将其转换为图像：

dot_data = StringIO()
export_graphviz(clf, out_file=dot_data,  
                filled=True, rounded=True,
                special_characters=True, feature_names = iris.feature_names,
                class_names=['setosa', 'versicolor', 'virginica'])
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())

这将生成一个PNG格式的决策树图像。