R语言NA替换函数运行时索引为数值而非NA的原因求解

原需求与错误代码 #

需求：编写函数，将向量中的所有NA值替换为向量内所有非NA数值的平均值。
你编写的代码如下：

na_replace <- function (x)
{for (i in 1:length(x)) 
   if (is.numeric(x[i]))
     {average<- c(is.numeric(x[i]))}
     if (is.na(x[i])) 
   {x[i] = mean(average)}
    return(x)}

测试输入为c(1,NA,3,NA)时，返回结果为c(1, NA, 3, 1)，与预期不符。

错误原因 #

• 循环体缺少大括号包裹：R中如果循环/条件判断后没有大括号，仅会把紧随其后的第一行代码识别为分支内逻辑。你的代码中for循环仅包含第一个if判断，第二个if (is.na(x[i]))完全在循环外，只会在循环结束后执行一次，因此只有最后一个元素位置的NA会被修改，第一个NA无处理逻辑。
• 平均值计算逻辑完全错误：你定义的average存储的是每次is.numeric(x[i])返回的逻辑值，数值类型元素对应TRUE（即1），非数值对应FALSE（即0），最终计算的mean(average)是逻辑向量的均值，和需求要求的「向量内非NA数值的平均值」完全无关。
• NA类型判定不符合预期：数值型向量中的NA属于数值型缺失值，is.numeric(NA)返回TRUE，因此你的第一个判断会把NA也识别为数值类型，不符合你原本想筛选非NA数值的逻辑。

正确实现 #

不需要写循环，直接用向量化操作即可完成需求，代码更简洁高效：

na_replace <- function(x) {
  # 提取所有非NA的数值计算均值
  valid_nums <- x[!is.na(x) & is.numeric(x)]
  avg <- mean(valid_nums)
  # 替换所有NA为均值
  x[is.na(x)] <- avg
  return(x)
}

调用na_replace(c(1,NA,3,NA))会返回预期结果c(1, 2, 3, 2)。

内容的提问来源于stack exchange，提问作者Merav Gold