python 实现KNN

# -*- coding: utf-8 -*-
from numpy import *
import operator
def createData():
    group=array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
    labels=['a','a','b','b']
    return group, labels

def classify(prex,inputx,labels,k):
    datasize=inputx.shape[0]
    # 将矩阵扩充
    prex_exp=tile(prex,(datasize,1))
    mat_diff=prex_exp-inputx
    dis_mat=mat_diff**2
    distance=sum(dis_mat,1)
    # 得到排序的index， 降序为argsort(-distance)
    ind_dis=argsort(distance)
    classCount={}
    print(ind_dis)
    for j in range(k):
        i=ind_dis[j]
        print(labels[i])
        # 开始直接classCount[labels[i]]=classCount[labels[i]]+1,但当标签还没开始计数时，不存在
        # get如果没找到，返回默认值
        classCount[labels[i]]=classCount.get(labels[i],0)+1
    print(classCount)  
    print('test')  
    # 字典排序 按key  e:e[0]， 按value e:e[0]  
    sortedClassCount=sorted(classCount.items(),key=lambda e:e[1],reverse=True)
    return sortedClassCount[0][0]
            

group,labels=createData()
print(classify([0,0],group,labels,3) )