使用 sapply 进行中值插补

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

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 中的任何 apply 函数获得相同的结果？

你可以使用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})

我确信如果您正在寻找的是性能，有人会提供数据表解决方案（不幸的是我不熟悉该包，所以我自己无法做到）。