R - Как векторизовать вычисление в 2d-матрице?

Question

Я хотел бы создать матрицу п х п размерности, со значениями

f = matrix(0,n,n) 
for (x in (1:n)) { 
    for (y in (1:n)) { 
     f[x,y] = x^2 + y^2 
    } 
} 

Есть ли способ векторизации это (чтобы избежать двух петель), чтобы ускорить программу?

Answer 1

Вы можете использовать outer

outer(x, y, function(x, y) x^2 + y^2) 

Например

x <- 1:4 
y <- 2:5 
outer(x, y, function(x, y) x^2 + y^2) 
    [,1] [,2] [,3] [,4] 
[1,] 5 10 17 26 
[2,] 8 13 20 29 
[3,] 13 18 25 34 
[4,] 20 25 32 41 

Answer 2

Как насчет этого?

n <- 10 
M <- matrix(rep(1:n,n), ncol=n, byrow=T) 
f <- (M**2)+t((M**2)) 
f