#-*- coding:utf-8 -*-
import numpy as np
import operator

def createDataset():
    #四組二維特征
    group = np.array([[5,115],[7,106],[56,11],[66,9]])
    #四組對應(yīng)標(biāo)簽
    labels = ('動作片','動作片','愛情片','愛情片')
    return group,labels

"""
KNN算法
"""
def classify(intX, dataSet, labels, k):
    '''
    numpy中shape[0]返回?cái)?shù)組的行數(shù)，shape[1]返回列數(shù)
    '''
    dataSetSize = dataSet.shape[0]

    """
    將intX在橫向重復(fù)dataSetSize次，縱向重復(fù)1次
    例如intX=([1,2])--->([[1,2],[1,2],[1,2],[1,2]])便于后面計(jì)算
    """
    diffMat = np.tile(intX, (dataSetSize, 1)) - dataSet

    """
    計(jì)算距離:歐式距離, 特征相減后乘方,然后再開方
    """
    sqdifMax = diffMat**2
    seqDistances = sqdifMax.sum(axis=1)
    distances = seqDistances**0.5

    #返回distance中元素從小到大排序后的索引
    print ("distances:",distances)
    sortDistance = distances.argsort()
    print ("sortDistance:", sortDistance)

    """
    取出前k個元素的類別
    """
    classCount = {}
    for i in range(k):
        voteLabel = labels[sortDistance[i]]
        s = "第{}個voteLabel={}".format(i, voteLabel)
        print(s)
        classCount[voteLabel] = classCount.get(voteLabel,0)+1

    #dict.get(key,default=None),字典的get()方法,返回指定鍵的值,如果值不在字典中返回默認(rèn)值。
    #計(jì)算類別次數(shù)

    #key=operator.itemgetter(1)根據(jù)字典的值進(jìn)行排序
    #key=operator.itemgetter(0)根據(jù)字典的鍵進(jìn)行排序
    #reverse降序排序字典
    sortedClassCount = sorted(classCount.items(), key = operator.itemgetter(1), reverse = True)
    #結(jié)果sortedClassCount = [('動作片', 2), ('愛情片', 1)]
    print ("sortedClassCount:")
    print(sortedClassCount)
    return sortedClassCount[0][0]

if __name__ == '__main__':
    group,labels = createDataset()
    test = [20,101]
    test_class = classify(test,group,labels,3)
    print (test_class)