具体操作
# 导入库：KNeighborsClassifier import numpy as np from sklearn.neighbors import KNeighborsClassifier import matplotlib.pyplot as plt from matplotlib.colors import ListedColormap # 导入sklearn自带数据集 from sklearn import datasets # 得到训练样本 iris = datasets.load_iris() X = iris.data[:,:2] y = iris.target #定义三种颜色代表三种蓝蝴蝶 cmap_species = ListedColormap(['#FF0000','#00FF00','#0000FF']) #c对应颜色池映射到target值 plt.scatter(X[:,0],X[:,1],c = y,cmap = cmap_species) plt.show()