Python:用sklearn中的DecisionTreeClassifier，根据iris数据生成决策树 并实现西瓜书 决策树

需求包分析：

关键词：sklearn、DecisionTreeClassifier、iris数据、决策树

对应包：sklearn.tree.DecisionTreeClassifier、sklearn.datasets.load_iris、matplotlib.pyplot

分析完毕，实现代码如下

import matplotlib.pyplot as plt
from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()
iris_tree = tree.DecisionTreeClassifier(criterion='entropy')
iris_tree = iris_tree.fit(iris.data,iris.target)
plt.figure(1)
tree.plot_tree(iris_tree)
plt.show()

运行结果：

思考

我们是否可以用上面的方法对其他数据集进行分析？

我们先将iris的文件导出，看看里面的格式

from sklea