python分类散点图_详解python scatter散点图用循环分类法加图例

python scatter散点图用循环分类法加图例

本文实例为大家分享了python scatter散点图用循环分类法加图例，供大家参考，具体内容如下

import matplotlib.pyplot as plt

import kNN

plt.rcParams['font.sans-serif']=['Simhei']

plt.rcParams['axes.unicode_minus']=False

datingDataMat, datingLabels = kNN.file2matrix('datingTestSet2.txt')

plt.figure()

type1_x = [] #一共有3类，所以定义3个空列表准备接受数据

type1_y = []

type2_x = []

type2_y = []

type3_x = []

type3_y = []

for i in range(len(datingLabels)): #1000组数据，i循环1000次

if datingLabels[i] == '1': #根据标签进行数据分类,注意标签此时是字符串

type1_x.append(datingDataMat[i][0]) #取的是样本数据的第一列特征和第二列特征

type1_y.append(datingDataMat[i][1])

if datingLabels[i] == '2':

type2_x.append(datingDataMat[i][0])

type2_y.appe