graphviz python

Python Graphviz是一个用于绘制图形的Python库，可以用它来画决策树。首先，需要安装Graphviz和Python Graphviz库。然后，可以使用Python代码来创建决策树，并使用Graphviz库将其可视化。具体步骤如下：

1. 安装Graphviz和Python Graphviz库

可以在Graphviz官网下载安装包，然后安装。安装完成后，可以使用pip安装Python Graphviz库：

pip install graphviz

2. 创建决策树

可以使用Python的scikit-learn库来创建决策树。以下是一个简单的例子：

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

iris = load_iris()
X = iris.data[:, 2:]
y = iris.target

tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)

3. 可视化决策树

使用export_graphviz函数将决策树导出为Graphviz格式：

export_graphviz(
        tree_clf,
        out_file=image_path("iris_tree.dot"),
        feature_names=iris.feature_names[2:],
        class_names=iris.target_names,
        rounded=True,
        filled=True
    )

然后使用Graphviz库将其可视化：

import graphviz

with open("iris_tree.dot") as f:
    dot_graph = f.read()
graphviz.Source(dot_graph)

这样就可以在Jupyter Notebook或其他Python环境中看到决策树的可视化结果了。