使用 sapply 的中值插补

Question

我想替换数据框列中的缺失值。我写了以下代码

MedianImpute <- function(data=data)
     {
      for(i in 1:ncol(data))
        {        
        if(class(data[,i]) %in% c("numeric","integer"))
          {
          if(sum(is.na(data[,i])))
            {
            data[is.na(data[,i]),i] <- 
                          median(data[,i],na.rm = TRUE)
            }
          }
        }
      return(data)
      }

这将返回 NA 被列中位数替换的数据帧。我不想使用 for 循环，如何使用 R 中的任何应用函数获得相同的结果？

Answer 1

您可以使用apply在所有列中应用函数

dat<-data.frame(c1=c(1,2,3,NA),c2=c(10, NA, 20, 30))
apply(dat, 2, function(x) ifelse(is.na(x), median(x, na.rm=T), x))

稍微快一点

imputeMedianv3<-function(x) apply(x, 2, function(x){x[is.na(x)]<-median(x, na.rm=T); x})

我敢肯定，如果您要寻找的是性能，那么有人会提供数据表解决方案（不幸的是，我不熟悉该软件包，因此我自己做不到）。