编写R函数：跨多数据框选择重命名指定变量并合并

现有同一跨国调查项目的多个数据框（如df_w1、df_w2），同一问题对应的变量在不同数据框中列名不同——比如变量X在df_w1中叫v2，在df_w2中叫V34。示例数据如下：

df_w1 <- data.frame(v2 = c(3,4,6))
df_w2 <- data.frame(V34 = c(2,7,9))

需要编写一个函数，满足以下要求：

• 传入数据框列表和目标变量的列名映射列表
• 自动识别各数据框中对应列，重命名为统一名称（如X）
• 合并为包含wave标识的新数据框，最终格式如下：

wave X
w1   3
w1   4  
w1   6
w2   2
w2   7
w2   9

• 支持为多个不同变量执行该操作

以下是符合需求的R函数，兼顾单个/多个变量的处理场景，同时保留参数灵活性：

merge_wave_vars <- function(df_list, var_mapping, wave_names = names(df_list)) {
  # 参数合法性校验
  stopifnot(length(df_list) == length(var_mapping), length(df_list) == length(wave_names))
  
  # 遍历处理每个数据框
  processed_dfs <- lapply(seq_along(df_list), function(i) {
    current_df <- df_list[[i]]
    current_map <- var_mapping[[i]]
    current_wave <- wave_names[i]
    
    # 提取指定列并重命名
    selected_cols <- current_df[, names(current_map), drop = FALSE]
    colnames(selected_cols) <- current_map
    
    # 添加wave标识列
    selected_cols$wave <- current_wave
    
    # 调整列顺序，将wave列置于首位
    selected_cols[, c("wave", names(current_map))]
  })
  
  # 合并所有处理后的数据框
  do.call(rbind, processed_dfs)
}

函数参数说明 #

• df_list：待处理的数据框列表，建议给列表元素命名（如list(w1 = df_w1, w2 = df_w2)），方便自动生成wave标识
• var_mapping：列名映射列表，每个元素是一个命名向量——键为原数据框中的列名，值为统一后的变量名。比如单个变量映射为list(c(v2 = "X"), c(V34 = "X"))，多变量映射为list(c(v2 = "X", v5 = "Y"), c(V34 = "X", V35 = "Y"))
• wave_names：可选参数，手动指定每个数据框对应的wave标识，默认使用df_list的元素名称

示例测试 #

单个变量处理 #

# 构造输入
df_list <- list(w1 = df_w1, w2 = df_w2)
var_mapping <- list(c(v2 = "X"), c(V34 = "X"))

# 调用函数
result <- merge_wave_vars(df_list, var_mapping)
print(result)

输出结果：

wave X
w1   w1 3
w1   w1 4
w1   w1 6
w2   w2 2
w2   w2 7
w2   w2 9

多个变量处理 #

如果需要同时处理多个变量，比如变量Y在df_w1中是v5，在df_w2中是V35：

# 构造多变量测试数据
df_w1_multi <- data.frame(v2 = c(3,4,6), v5 = c(10,20,30))
df_w2_multi <- data.frame(V34 = c(2,7,9), V35 = c(40,50,60))

# 配置输入参数
df_list_multi <- list(w1 = df_w1_multi, w2 = df_w2_multi)
var_mapping_multi <- list(c(v2 = "X", v5 = "Y"), c(V34 = "X", V35 = "Y"))

# 执行函数
result_multi <- merge_wave_vars(df_list_multi, var_mapping_multi)
print(result_multi)

输出结果：

wave X  Y
w1   w1 3 10
w1   w1 4 20
w1   w1 6 30
w2   w2 2 40
w2   w2 7 50
w2   w2 9 60

内容的提问来源于stack exchange，提问作者T.E.G.