эффективная векторизация двойного цикла для цикла

Question

как можно векторизовать следующий двойной цикл в R?

a <- seq(1,10, length=5) 
b <- seq(0,1, length=4) 
fun <- function(a,b){return(a+b)} 

out <- matrix(NaN, nrow=5, ncol=4) 

    for(i in 1:5) { 
    for(j in 1:4) { 
     out[i, j] <- fun(a[i], b[j]) 
    } 
    } 

Я попытался, например, безуспешно. Пожалуйста, посоветуйте, заранее спасибо

outer(1:nrow(out), 1:ncol(out), FUN = fun(a,b)) 
mapply(out, fun) 

Answer 1

насчет:

outer(a, b, '+') 

## > outer(a, b, '+') 
##  [,1]  [,2]  [,3] [,4] 
## [1,] 1.00 1.333333 1.666667 2.00 
## [2,] 3.25 3.583333 3.916667 4.25 
## [3,] 5.50 5.833333 6.166667 6.50 
## [4,] 7.75 8.083333 8.416667 8.75 
## [5,] 10.00 10.333333 10.666667 11.00 

Answer 2

По общему правилу, векторизации самого внутреннего цикла, как правило, достаточно, чтобы получить вам выигрыш в производительности. В этом случае это будет означать:

for(i in 1:5) { 
    out[i,]<- a[i] + b 
} 

Предполагая, что a и b оба имеют длину О (п), то, делая это таким образом, вы убедитесь, что накладные расходы от переборе цикла в R только понесены O (n) раз, а не O (n^2) раз, если вы выполняете обе петли в R. Так как время выполнения алгоритма будет равно O (n^2), то дополнительные O (n) t сильно влияет на время работы.

Конечно, для этого конкретного примера вы можете использовать outer, как и другие.