#[3,1],[3,2] 类别1
#[1,3],[1,4] 类别2
#[5,2]
#把距离[5,2]距离最近的前两个点的类别找出
import numpy as np
注释:c1是属于类别1的
c1 = np.array([3,1],[3,2])
注释:c2是属于类别2的
c2 = np.array([1,3],[1,4])
注释:判断t分别到类别1中的距离和到类别2中的距离
t = np.array([5,2])
注释:以列相减求两点之间的距离
dis_a = np.linalg.norm(t-c1,axis=1)
注释:以列相减求两点之间的距离
dis_b = np.linalg.norm(t-c2,axis=1)
注释:给dis_a和dis_b排序相减
dis_c = np.sort(dis_a) - np.sort(dis_b)
注释:判断dis_c的长度,是表示判断离目标点的距离比较近的点的有几个
len(dis_c[dis_c<0])