使用决策树对鸢尾花进行分类python_决策树实现鸢尾花的分类

鸢尾花数据集介绍

一：读取数据

from sklearn.datasets import load_iris
# 导入数据集Iris
iris = load_iris() #导入数据
iris_feature= iris.data #特征数据
iris_target = iris.target #分类数据
# print(iris.data)  # 输出数据
1
2
3
4
5
6

print(type(iris))
print(type(iris_feature))
print(type(iris_target))  # numpy 数据类型
1
2
3

<class 'sklearn.utils.Bunch'>
<class 'numpy.ndarray'>
<class 'numpy.ndarray'>
1
2
3

# numpy数据查看--索引
print(iris_feature[2])
print(iris_feature[2,1])
1
2
3

[4.7 3.2 1.3 0.2]
3.2
1
2

二：鸢尾花类别

target介绍

# print(iris.data)  # 输出数据
print(iris.target) #输出真实标签
print(len(iris.target)) #样本个数
print(iris.data.shape)
1
2
3
4

[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
(150, 4)
1
2
3
4
5
6
7

三：数据可视化

1：绘制直方图

import pandas as pd
1