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(可迭代对象)返回。

上一篇下一篇

猜你喜欢

热点阅读