机器学习实战--KNN

import numpy as np
import operator
def createDataSet():
    group = np.array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
    labels = ['A','A','B','B']
    return group,labels

def classify0(inX,dataset,labels,K):
    datasetsize = dataset.shape[0]
    #返回一个5行一列的矩阵，每行和INX相同
    diffMat = np.tile(inX,(datasetsize,1)) - dataset
    #矩阵求平方是对其中每一个元素求平方
    sqDiffMat = diffMat**2
    #axis=1表示对列求和
    sqDistance = sqDiffMat.sum(axis=1)
    distance = sqDistance**0.5
    #返回按值从小到大排列的原始下标数组
    sortedDistIndicies = distance.argsort()
    classCount = {}
    for i in range(K):
        voteLable = labels[sortedDistIndicies[i]]
        classCount[voteLable] = classCount.get(voteLable,0) + 1
    #将字典元素按照第二项排序，sorted默认递增排序，reverse取反
    sortedClassCount = sorted(classCount.items(),
                              key = operator.itemgetter(1),reverse = True)
    return sortedClassCount[0][0]


if __name__ == '__main__':
    group,labels = createDataSet()
    labelpredict = classify0([0.5,0.6],group,labels,2)
    print(labelpredict)