用R中的Lapply族替换双嵌套的for循环

Question

我目前正在使用double（嵌套） for循环来实现以下公式： 。

一切正常。 但是我想知道是否可以：

（A）输入r如在本情况下，单个数.3 ，而不是矩阵.3秒和1 S'

（B）使用lapply系列代码，例如mapply而不是嵌套for -loop？

V <- c(.25, .5, .75) 
m <- length(V)

r <- matrix(c(1, .3, .3, .3, 1, .3, .3,.3, 1), 3, 3)

sumV <- 0 

for (i in 1:nrow(r)) {
  for (j in 1:nrow(r)) {
    sumV <- sumV + r[i,j] * sqrt(V[[i]]*V[[j]])
     }
   } 

(1/m)^2 * sumV # Final answer

Answer 1

您尝试的一个简短版本是

(1/m)^2 * sum(sqrt(outer(V, V)) * r)
#[1] 0.2599292

outer将每个元素与每个其他元素相乘，这就是double循环正在做的事情。 然后，我们采取sqrt所有的值，用乘法r矩阵， sum它们并通过乘以(1/m)^2 。

---

我们想将对角元素乘以1，将其余的元素与r值相乘，这在r是矩阵时很容易，但是，如果它是单个数字，则需要相应地构造矩阵。

r <- .3
mat <- sqrt(outer(V, V))
mult_matrix <- matrix(r, ncol(mat), nrow(mat))
diag(mult_matrix) <- 1
(1/m)^2 * sum(mat * mult_matrix)
#[1] 0.2599292