Comment appliquer une fonction sur les indices de chaque élément de la matrice

Question

Je me demande s'il existe une fonction intégrée dans R qui applique une fonction à chaque élément de la matrice (bien sûr, la fonction doit être calculée en fonction des indices de la matrice). L'équivalent serait quelque chose comme ceci :

 matrix_apply <- function(m, f) {
  m2 <- m
  for (r in seq(nrow(m2)))
    for (c in seq(ncol(m2)))
      m2[[r, c]] <- f(r, c)
  return(m2)
}

S'il n'y a pas une telle fonction intégrée, quelle est la meilleure façon d'initialiser une matrice pour qu'elle contienne des valeurs obtenues en calculant une fonction arbitraire qui a des indices de matrice comme paramètres ?

Answer 1

Vous pensez peut-être à outer :

 rows <- 1:10
cols <- 1:10

outer(rows,cols,"+")

      [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] [,10]
 [1,]    2    3    4    5    6    7    8    9   10    11
 [2,]    3    4    5    6    7    8    9   10   11    12
 [3,]    4    5    6    7    8    9   10   11   12    13
 [4,]    5    6    7    8    9   10   11   12   13    14
 [5,]    6    7    8    9   10   11   12   13   14    15
 [6,]    7    8    9   10   11   12   13   14   15    16
 [7,]    8    9   10   11   12   13   14   15   16    17
 [8,]    9   10   11   12   13   14   15   16   17    18
 [9,]   10   11   12   13   14   15   16   17   18    19
[10,]   11   12   13   14   15   16   17   18   19    20

C'est clairement un exemple de fonction assez trivial, mais vous pouvez également fournir votre propre fonction personnalisée. Voir ?outer .

Éditer

Contrairement au commentaire ci-dessous, vous pouvez également utiliser outer avec des fonctions non vectorisées en... les vectorisant !

 m <- matrix(1:16,4,4)

#A non-vectorized function 
myFun <- function(x,y,M){
     M[x,y] + (x*y)
}

#Oh noes! 
outer(1:4,1:4,myFun,m)
Error in dim(robj) <- c(dX, dY) : 
  dims [product 16] do not match the length of object [256]

#Oh ho! Vectorize()! 
myVecFun <- Vectorize(myFun,vectorize.args = c('x','y'))

#Voila! 
outer(1:4,1:4,myVecFun,m)
     [,1] [,2] [,3] [,4]
[1,]    2    7   12   17
[2,]    4   10   16   22
[3,]    6   13   20   27
[4,]    8   16   24   32