Buscando un ifelse más eficiente()

Question

Mientras R ifelse es increíblemente útil, tiene un defecto en particular: en la llamada ifelse(test, yes, no) se evalúan todos los elementos de yes y no, incluso aquellos que se desecha.

Esto es más bien un desperdicio si se está usando en el medio de un ejercicio numérico complicado, dicen en una función que se alimenta a integrate, uniroot, optim o lo que sea. Por ejemplo, uno podría haber

ifelse(test, f(x, y, z), g(a, b, c)) 

donde f y g son arbitrariamente funciones complejas o lentos, posiblemente con la participación más anidados ifelse 's.

¿Alguien ha escrito un reemplazo para ifelse que solo evalúa los elementos de yes/no que se conservarán? Básicamente, algo en la línea de

out <- test 
for(i in seq_along(out)) 
{ 
    if(test[i]) out[i] <- f(x[i], y[i], z[i]) 
    else out[i] <- g(a[i], b[i], c[i]) 
} 

pero sin la torpeza/ineficacia de un bucle explícito. ¿Esto es posible incluso sin entrar en las entrañas de R?

Answer 1

No creo que el problema sea ifelse. f y g solo se evalúan una vez en su expresión. Creo que su problema es que f y g son lentos con los vectores.

Puede cambiar las llamadas a f y g para que solo se evalúen en un subconjunto del vector.

out <- numeric(length(test)) #or whatever the output type is 
out[test] <- f(x[test], y[test], z[test]) 
out[!test] <- g(x[!test], y[!test], z[!test]) 

Tendrá que ajustar si cualquier elemento de test son NA.