vectoriser une fonction R

Question

J'ai la fonction suivante

f <- function(x){sum(g(x - X))} 

où

X - n-dimensional vector with some data 
g is some vecrtorized function 

Comment puis-je vectoriser la fonction f afin qu'il puisse prendre d'entrée n dimensions et le rendement de sortie n dimensions?

je tentais les éléments suivants

rowSums(sapply(x, "-", X)) 

Le problème avec cette approche est qu'elle ne couvre pas le cas d'une dimension x. Est-il possible de couvrir les deux cas? Par exemple nous

x <- c(1,2,3) 
X <- c(6,9,1) 
g <- function(x){x^2} 

Si je sapply le code à base je reçois la réponse correcte (vecteur n dimensions)

rowSums(sapply(x, "-", X)) 
[1] -12 -21 3 

Mais si je mets x=1 et exécuter le même code, je reçois le mauvais réponse (vecteur n dimensions au lieu de scalaire)

rowSums(sapply(x, "-", X)) 
[1] -5 -8 0 

Ce n'est pas surprenant, puisque rowSums appliqué à la colonne vecteur donne le vecteur de colonne. Ce dont j'ai besoin dans le cas de unidimensionnelle, cependant, est d'appliquer sum. Existe-t-il un moyen élégant de le faire sans utiliser le if sous réserve de dimensions?

Answer 1

Il semble que vous avez deux exemples, je les ai nommés f et h:

X <- c(6,9,1) 

g <- function(x){x^2} 
f <- function(x){sapply(x,function(x)sum(g(x - X)))} 

f(1) 
# [1] 89 
f(1:3) 
# [1] 89 66 49 

h <- function(x){colSums(sapply(x,function(x) x-X))} 
h(1) 
# [1] -13 
h(1:3) 
# [1] -13 -10 -7 

Il ressemblait à colSums est ce que vous cherchez là-bas.