lundi 14 décembre 2020

Wrapper function with if conditions

Could someone please tell me how to combine wrapper function with if..else conditions? For example this wrapper:

wrapper<-function(x){
  varcoef<-function(x){
    sd(x)/mean(x)
  }
  apply(x,MARGIN = 2, FUN=varcoef)
}

wrapper(mtcars) 

With:

if(is.matrix(x)==TRUE){
  apply(x,2,function(x) sd(x)/mean(x))
} else if (is.data.frame(x)==TRUE){
  apply(x,2,function(x) sd(x)/mean(x))
} else print(NULL)

Thank you!

Aucun commentaire:

Enregistrer un commentaire