在 R 中的 knn 中实现距离矩阵

Question

我想计算 R 中的距离矩阵（使用欧几里得距离）

我知道如何在 for 循环中做到这一点

但是我如何使用 apply family fucntion 来做到这一点？

require(magrittr)

data(iris)

set.seed(1)
data <- iris[sample(nrow(iris))   ,] # 打散資料
Xtrain <- data[1:100,1:4]
Xtest <- data[101:150,1:4]

M <- nrow(Xtrain)
N <- nrow(Xtest)

distmatrix <- matrix(0,nrow = M,ncol = N)

for(i in 1:M){
  for(j in 1:N){
    distmatrix[i,j]<- sum((Xtrain[i,]-Xtest[j,])^2) %>% sqrt()
  }

}

Answer 1

考虑mapply与expand.grid为所有组合M和N ：

params <- expand.grid(M=seq(M), N=seq(N))

distmatrix2 <- matrix(mapply(function(i,j) sqrt(sum((Xtrain[i,]-Xtest[j,])^2)), 
                      params$M, params$N),
                      nrow = M, ncol = N)

# EQUIVALENT TO OP's distance 
identical(distmatrix, distmatrix2)
# TRUE