博雅数据机器学习09

博雅数据机器学习09

K近邻算法

## 声明储存K个邻居的index的列表

index_list = []

 

## 遍历每个测试样本的与训练集的举例，item类型为dict

for item in distances:

 

  

    ## item的类型为dict，即{'index': distance}

    ## 使用sorted函数对item按照distance进行升序排列,并取前10个距离最小的样本

    sorted_item = sorted(item.items(), key = lambda x: x[1])[0:10]

    

    ## sorted_item形如[(index, distance),(index, distance)...(index, distance)]

    ## 使用zip函数从sorted_items中获取对应的index

    indexes = list(zip(*sorted_item))[0]

    

    ## 将indexes加入到index_list中

    index_list.append(indexes)

index_list