鸢尾花数据集

先下载模块：

sudo python3 -m pip install --user numpy scipy matplotlib ipython jupyter pandas sympy
sudo python3 -m pip install -U scikit-learn

调用包：

from sklearn.datasets import load_irisfrom sklearn.ensemble

载入数据集：

iris = load_iris()

iris包含两个属性：iris.data和iris.target

data是一个矩阵，每一列代表一个特征，共四列150行，每行是被测样本

打印iris.data

[[5.1 3.5 1.4 0.2]

[4.9 3. 1.4 0.2]

[4.7 3.2 1.3 0.2]

[4.6 3.1 1.5 0.2]]

target代表每个样本实际属于哪个鸢尾属，0为山鸢尾(Iris-setosa)、1为杂色鸢尾(Iris-versicolor)、2为维吉尼亚鸢尾(Iris-virginica)

打印iris.target

[0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0

0 0 0 0 0 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1

1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 2

2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2

2 2]

可以看到总共150个样本，每个亚属有50个样本。

用matplotlib.pyplot将分类图画出，可以看到有明显的分类特点，因此适合用于做决策树训练集，这里就不贴代码了。