import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
from sklearn.neighbors import KNeighborsClassifier
from sklearn import datasets
step 2:数据导入
#使用莺尾花数据集的前两维数据,便于数据可视化
iris = datasets.load_iris() #加载鸢尾花数据集
X = iris.data[:, :2] #取鸢尾花数据集前两列特征
y = iris.target #y为鸢尾花的类别信息