将多个for循环转换为单个或嵌套的apply语句R

Question

我希望减少运行 multiple/nested 循环的过多处理时间。关于这个主题似乎有几个问题，但 R 的答案并不多，所以这里有另一个（希望更简单的）例子。

本质上，我们可以将此示例视为根据逻辑回归输出创建预测。我想将下面的 multiple/nested 循环转换为 apply 语句。

使用循环的示例：

set.seed(1111)

#Data
b0 <- matrix(c(rnorm(10,-1, 0.1), rnorm(10,-2, 0.1)), c(10,2))
b1 <- matrix(c(rnorm(10,-.4, 0.1), rnorm(10,-.1, 0.1)), c(10,2))
b2 <- matrix(c(rnorm(10, 0, 0.1), rnorm(10,-.1, 0.1)), c(10,2))
x <- rnorm(5, 0.5, 0.01)
x2 <- rnorm(5, 0.25, 0.01)

#Predictions Array
pred.ray = array(NA, dim = c(10, 5, 2))

# Can do for loops but takes waaaay too long on full data set ~ 20hrs
for(i in 1:5){
  for (j in 1:2){
    pred.ray[,i,j] = plogis(b0[,j] + b1[,j]*x[i] + b2[,j]*x2[i])
  }}

pred.ray

Answer 1

len_x <- 5
ncol_b <- 2
val <- len_x * ncol_b

result <- mapply(function(i, j) plogis(b0 + b1 * i + b2 * j), x, x2)
inds <- lapply(0:1, function(i) sort(rep(seq(i, val+i-1, ncol_b) * val, val) + 
rep(1:val, each=len_x)))
ans <- array(c(sapply(inds, function(i) result[i])), dim=c(10,5,2))
identical(pred.ray, ans)
# TRUE

将多个for循环转换为单个或嵌套的apply语句R

Converting multiple for loops to single or nested apply statements R

for-loop

r

apply

nested-loops

multidimensional-array