机器学习入门-加载sklearn中数据并用matplotlib进行可视化

from sklearn import datasets
import matplotlib.pyplot as plt


def get_data():
    """
    从sklearn中获取鸢尾花的数据
    :return:  鸢尾花数据的字典，字典中包括的key有：【'data', 'target', 'target_names', 'DESCR', 'feature_names']
    简单介绍一下： data就是（150， 4）的数据集，target表示1位数组，数字0～2表示分类， target_names表示分类名，DESCR表示对数据的描述
    feature_names: 特征值名称
    """
    iris = datasets.load_iris()
    return iris


def draw_graph(iris_data):
    """
    获取两个维度的数据进行数据可视化，由于鸢尾花共有4个特征，在平面中只能绘制2个特征，所以获取特征1和特征2进行绘制
    :param: 鸢尾花的数据集
    :return:
    """
    X = iris_data.data[:, :2]
    target_names = iris_data.target_names
    print(target_names)
    print(iris_data.feature_names )
    y = iris_data.target
    plt.scatter(X[y == 0, 0], X[y == 0, 1], color='red'