用于清洁R中的NA的循环

Question

有没有办法更有效地编写此代码？

for (k in 1:ncol(tr)){
  temp=tr[[k]]
  temp[is.na(temp)] = mean(temp[!is.na(temp)])
  tr[[k]]=temp
}

Answer 1

要替换矩阵或数据帧的for循环，apply通常是一个好主意：

apply(tr, 1,  function(temp) {temp[is.na(temp)] <- mean(temp, na.rm=TRUE); temp})

有关apply的更多信息 - 类似函数here。

Answer 2

sapply(tr, function(x) { 
  x[is.na(x)] <- mean(x, na.rm=TRUE)
  return(x) 
})