用python打印出决策树

以下是使用Python打印决策树的示例代码：

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

# 加载鸢尾花数据集
iris = load_iris()
X = iris.data
y = iris.target

# 训练决策树模型
tree_clf = DecisionTreeClassifier(max_depth=2)
tree_clf.fit(X, y)

# 将决策树输出为Graphviz格式
dot_data = export_graphviz(tree_clf,
                           out_file=None,
                           feature_names=iris.feature_names,
                           class_names=iris.target_names,
                           rounded=True,
                           filled=True)

# 使用Graphviz绘制决策树
graph = graphviz.Source(dot_data)
graph.render("iris_decision_tree")

运行代码后，会在当前目录下生成名为“iris_decision_tree”的文件，该文件就是绘制出的决策树图形。