How to write a custom function which will replace all the missing values in a vector with the mean of values in R

Question

I want to write a custom function which will replace all missing values in a vector with the calculated mean values in R

Answer 1

Consider this vector:

a<-c(1,2,3,NA,4,5,NA,NA)

Write the function to impute the values:

mean_impute<-function(x){
ifelse(is.na(x),mean(x,na.rm = T),x)
}

The result would be as follows:

mean_impute(a)

1 2 3 3 4 5 3 3