2.2.2(kNN分类算法)
2018-08-11 本文已影响0人
tmax
from numpy import * #注意import方式
import matplotlib.pyplot as plt
import kNN
datingDataMat,datingLabels=kNN.file2matrix('C:\Users\\tamx\Desktop\MLiA_SourceCode\machinelearninginaction\Ch02\datingTestSet2.txt')
plt.figure().add_subplot(111).scatter(datingDataMat[:,1],datingDataMat[:,2])
plt.show()
plt.figure().add_subplot(111).scatter(datingDataMat[:,1], datingDataMat[:,2], s=15.0*array(datingLabels),c=15.0*array(datingLabels))
#无需转换
#plt.figure().add_subplot(111).scatter(datingDataMat[:,1], datingDataMat[:,2], s=15.0*array(map(int,datingLabels)),c=15.0*array(map(int,datingLabels)))
plt.show()
scatter()
见https://www.cnblogs.com/shanlizi/p/6850318.html
map()
函数接收两个参数,一个是函数,一个是可迭代对象,如列表,字符串等,map将传入的函数依次作用到序列(可迭代对象)的每个元素,并把结果作为新的Iterator(可迭代对象)返回。