r - 在 R 中向量化双循环

我是 R 的新手，正在尝试对下面的代码进行矢量化。

什么是更好的方法来做到这一点？非常感谢!

*

l_mat <- data.frame(matrix(ncol = 4, nrow = 4))
datax <- data.frame("var1"= c(1,1,1,1), "Var2" = c(2,2,2,2), "Var3"=c(3,3,3,3), "Var4"=c(4,4,4,4))
for (i in 1:4) {
      for (j in 1:4) {
             if (datax[i, 2] == datax[j, 2]) {
                   l_mat[i, j] <- 100
               } else {
                     l_mat[i, j] <- 1
                 }
         }
   }

*

最佳答案

使用 outer 可以更好地完成。当我们检查第二列中的所有值是否与自身相匹配时，使用 outer 创建逻辑表达式，将其转换为数字索引，然后将这些值替换为 1 或 100

out <- 1 + (outer(datax[,2], datax[,2], `==`))
out[] <- c(1, 100)[out]

或者在一行中

ifelse(outer(datax[,2], datax[,2], `==`), 100, 1)

或者使用 pmax 和 outer 的变体

do.call(pmax, list(outer(datax[,2], datax[,2], `==`) * 100, 1))

关于r - 在 R 中向量化双循环，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/52501794/