R语言函数内用中位数替换数据框NA不生效，外部运行正常如何解决

• 核心问题1：函数未返回修改后的数据框，R默认按值传递参数
  R中函数内部修改传入的data.frame/tibble时，默认操作的是原对象的副本，函数执行结束后副本会被销毁，不会修改全局环境中的main_df对象。你手动执行时是直接修改全局的main_df，所以能生效，封装成函数后没有返回修改结果自然看不到变化。
• 问题2：类型判断逻辑冗余且存在错误
  typeof()对于数值型变量只会返回integer或double，不存在返回numeric的情况，所以你写的col_type_vector[item] == "numeric"永远为假，属于无效判断。
• 问题3：NA索引提取逻辑存在隐性风险
  data_frame[column_name]返回的是单列tibble而非向量，is.na()作用在tibble上返回的是矩阵格式的逻辑值，which()提取的索引可能出现不符合预期的情况，建议用[[提取列向量来判断NA位置。

首先修改函数，添加return返回修改后的数据框，同时优化逻辑：

set_na_to_median <- function(data_frame, column_name) {
  # 提取列向量判断NA
  col_vec <- data_frame[[column_name]]
  median_value <- median(col_vec, na.rm = TRUE)
  col_vec[is.na(col_vec)] <- median_value
  data_frame[[column_name]] <- col_vec
  # 返回修改后的数据框
  return(data_frame)
}

col_type_vector <- sapply(main_df, typeof)
# 从第3列开始遍历
for (item in names(col_type_vector)[3:length(col_type_vector)]) {
  # 简化数值类型判断
  if (col_type_vector[item] %in% c("integer", "double")) {
    # 将函数返回的结果赋值回main_df
    main_df <- set_na_to_median(main_df, item)
  }
}

如果是操作tibble，直接用dplyr语法可以避免自定义函数和循环，代码更简洁：

library(dplyr)

main_df <- main_df %>%
  # 从第3列开始，对所有数值列替换NA为中位数
  mutate(across(3:last_col() & where(is.numeric), 
                ~replace(., is.na(.), median(., na.rm = TRUE))))

内容的提问来源于stack exchange，提问作者Fazal Mahmud Niloy