r语言编写一个函数满足条件

如图所示

我先说一下我对这个问题的理解
首先要提取列向量
然后再用for循环结构和if控制结构判断每个向量是否=NA，等于就替换，不等直接+1判断下一个向量
想法很美好，但不知道怎么搞…

replace_NA_with_sample <- function(df) {
  # 创建一个新的数据框，用于存储替换缺失值后的结果
  df2 <- data.frame()
  
  # 遍历每一列
  for (col_name in colnames(df)) {
    # 提取当前列的向量
    col_vec <- df[[col_name]]
    
    # 判断当前列是否存在缺失值
    if (sum(is.na(col_vec)) > 0) {
      # 抽样非缺失值，并用抽样值替换缺失值
      col_vec[is.na(col_vec)] <- sample(col_vec[!is.na(col_vec)], sum(is.na(col_vec)), replace = TRUE)
    }
    
    # 将当前列添加到新数据框中
    df2[[col_name]] <- col_vec
  }
  
  # 返回替换缺失值后的数据框
  return(df2)
}