如何将数据框每行最后一个非NA值替换为另一列的值

需求说明 #

给定数据集，需要将每行从左到右的最后一个非NA值替换为指定列（如示例中的E列）的值，同时需要一个通用函数来适配不同列数的数据框。

示例数据集 #

tdf <- data.frame(
    A = c(1, 2, 5, 4),
    B = c(NA, 3, 4, 5),
    C = c(NA, NA, NA, NA),
    D = c(NA, NA, NA, NA),
    E = c(10, 20, 30, 40))

目标结果 #

enddf <- data.frame(
    A = c(10, 2, 5, 4),
    B = c(NA, 20, 30, 40),
    C = c(NA, NA, NA, NA),
    D = c(NA, NA, NA, NA),
    E = c(10, 20, 30, 40))

问题分析 #

你之前尝试的代码硬编码了A、B列的判断逻辑，既无法适配不同列数的数据框，也没有正确定位每行最后一个非NA值的位置，因此方向有误。

通用解决方案 #

方案1：使用tidyverse工具链 #

以下函数支持任意列数的数据框，只需传入目标列名即可：

library(dplyr)
library(tidyr)

replace_last_non_na <- function(df, target_col) {
  # 筛选出除目标列外的所有待检查列
  cols_to_check <- setdiff(names(df), target_col)
  
  df %>%
    rowwise() %>%
    mutate(
      # 定位当前行最后一个非NA的列名
      last_non_na_col = tail(cols_to_check[!is.na(c_across(all_of(cols_to_check)))], 1),
      # 仅对最后一个非NA列进行值替换
      across(all_of(cols_to_check), ~ if_else(cur_column() == last_non_na_col, !!sym(target_col), .x))
    ) %>%
    ungroup() %>%
    select(-last_non_na_col) # 移除临时辅助列
}

# 测试函数
enddf <- replace_last_non_na(tdf, "E")

方案2：使用Base R实现 #

如果不想依赖tidyverse包，可以用Base R编写通用函数：

replace_last_non_na_base <- function(df, target_col) {
  cols_to_check <- setdiff(names(df), target_col)
  target_vals <- df[[target_col]]
  
  # 逐行处理
  for (i in 1:nrow(df)) {
    row_data <- df[i, cols_to_check, drop = FALSE]
    # 获取当前行非NA值的列索引
    non_na_positions <- which(!is.na(row_data))
    if (length(non_na_positions) > 0) {
      # 取最后一个非NA列的位置并替换值
      last_pos <- tail(non_na_positions, 1)
      df[i, cols_to_check[last_pos]] <- target_vals[i]
    }
  }
  return(df)
}

# 测试函数
enddf_base <- replace_last_non_na_base(tdf, "E")

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