# 鸢尾花的案例学习
#  1. 导入小型的数据
from  sklearn.datasets import load_iris
import numpy as np
import  pandas as pd
import  seaborn as sbn
import matplotlib.pyplot as plt
#  2. 获取数据
iris=load_iris()
# 3.查看数据

print("数据集\n ",len(iris.data))
print("特征名字\n",iris.target_names)
print("目标\n",iris.target)
#  4.绘制初始图形   使用seaborn Seaborn是封装好的matplib的api
'''
使用 seaborn 还需要使用 pandas ，因为seaborn传入的是pandas的数据格式
'''
# 4.1 将数据转化成dataframe 的格式
irisDataFrame = pd.DataFrame(iris.data,columns=["speal_length","speal_width","petal_length","petal_width"])
# 4.2给 irisDataFrame 添加一列  种类  并且赋值为 target
irisDataFrame["species"]=iris.target
# 4.3 编写绘图函数  花的数据资料   x轴  y轴
def iamge_iris(irisdata,column1,column2):
    sbn.lmplot(x=column1,y=column2,data=irisdata,hue="species",fit_reg=True)
    plt.xlabel(column1)
    plt.ylabel(column2)
    plt.title("鸢尾花的一个图形")
    plt.show()
iamge_iris(irisDataFrame,"speal_length","speal_width")