鸢尾花数据集

Posted 2021-01-12 nuan-z

1、2、3

#加载numpy包
import numpy
#加载sklearn包
from sklearn.datasets import load_iris 
#读出鸢尾花数据集data
data=load_iris()
#查看data类型
print(type(data))
#查看数据内容
print(data.keys())

4、

#鸢尾花的四个特征
data_feature= data.feature_names
iris_data=data.data
print(data_feature)
print(iris_data)
#鸢尾花的三个类别
data_target =data.target_names
print(data_target)
iris_data=data.target
print(iris_data)
#数据类型
type(iris_data)

5、

#鸢尾花花萼长度的数据
sepal_length=numpy.array(list(len[0] for len in data[‘data‘]))
print(sepal_length)

6、

#鸢尾花花瓣长度的数据
petal_length=numpy.array(list(len[2] for len in data[‘data‘])) 
print(petal_length)

#鸢尾花花瓣宽度的数据
petal_width=numpy.array(list(len[3] for len in data[‘data‘])) 
print(petal_width)

7、

#取出某朵花的四个特征
print(data.data[0])
#取出某朵花的类别
print(data.target_names[0])

8、9

#定义三个列表来存放不同类型花朵的类别
setosa_data = []
versicolor_data = []
virginica_data = []
#
for i in range(0,150):
    #生成为setosa类的鸢尾花花数据
    if  data.target[i] == 0: 
        data1 = data.data[i].tolist()
        data1.append(‘setosa‘)
        setosa_data.append(data1)
    #生成为versicolor类的鸢尾花数据  
    elif data.target[i] == 1: 
        data1 = data.data[i].tolist()
        data1.append(‘versicolor‘)
        versicolor_data.append(data1)
    #剩下的为virginica类的鸢尾花数据  
    else:                         
        data1 = data.data[i].tolist()
        data1.append(‘virginica‘)
        virginica_data.append(data1)
#生成新的数组，每个元素包含四个特征+类别
newdata=(setosa_data ,versicolor_data,virginica_data)
print(newdata)