R-2个矩阵的“应用”

Question

我正在做R赋值，并且我必须编写一个函数来做dist.xyz所做的事情。

dist.xyz(a, b = NULL, all.pairs=FALSE) 

A和b是数字矩阵，该函数计算‘a’和‘b’的相应行之间的距离。

我尝试了一个for循环(如下所示)，但它花费的时间太长，“应用”一次只允许我们对一个矩阵进行操作。

 dis = vector()
 for (i in 1:nrow(a)) {
    append(dis,sqrt(sum((a[i,] - b[i,]) ^ 2)))
 } 

有什么方法可以“应用”到两个矩阵上吗？

提前感谢

Answer 1

如果您有示例数据，将会更容易。但这是我的观点。这不是“应用于2个矩阵”的通用解决方案。然而，在你的例子中，你只需要申请一个矩阵a-b，因为每一行的元素差异是你首先要做的。然后对每一行应用平方、求和和平方根，以获得结果。

set.seed(7)   # just to ensure reproducible results

rowDist<-function(a,b) {
   apply(a-b,1,function(x)sqrt(sum(x^2)))
}

a<-matrix(rnorm(25),5,5)
b<-matrix(rnorm(25),5,5)

rowDist(a,b)
#[1] 2.716251 2.685056 3.699462 2.125998 3.437412