使用R仅查找一个点(不是整个矩阵)的k-最近邻居

时间:2012-12-24 19:46:40

标签: r knn

是否有一个包或一种简单的方法来使用R来搜索一个点的k-最近邻居(特别是kd树)? 提供此功能的所有软件包(例如RANN或FNN ...)计算矩阵中所有点的knn,我只需要一个点。

例如,我有一个10点“A”到“E”的矩阵,我想在“A”找到4个其他点(“B”到“E”)之间的2个最近邻居而不做同样的事情计算数据集中的所有行(不计算“B”,“C”,“D”,“E”的knn)

我希望我的问题很明确,我的英语不好。

感谢您的帮助,

1 个答案:

答案 0 :(得分:13)

如果我理解正确,您可以使用FNN包执行此操作:

> library(FNN)
> X <- matrix(runif(100), 5, 5)
> X
          [,1]      [,2]      [,3]      [,4]      [,5]
[1,] 0.7475301 0.6725876 0.2511358 0.5048512 0.1196027
[2,] 0.5777907 0.6337206 0.8334608 0.5067914 0.6410024
[3,] 0.5488786 0.9613076 0.2217271 0.6906149 0.7396482
[4,] 0.8230380 0.8596784 0.6348114 0.6211107 0.3089131
[5,] 0.6531433 0.8682462 0.2555402 0.2443061 0.5292509
> knnx.dist(X[-1,], X[1, , drop=FALSE], k=2)
          [,1]     [,2]
[1,] 0.4870996 0.531889
> knnx.index(X[-1,], X[1, , drop=FALSE], k=2)
     [,1] [,2]
[1,]    3    4

请注意,knnx.index的结果与传递给函数的矩阵有关,因此3和4实际上意味着第4和第5行是原始数据集。