R语言如何批量处理多数据框设首行为列名，避免重复if语句

示例说明 #

现有如下两个dataframe：

DF1 = data.frame(V1 = c("","A", "B"), V2 = c("x",0,1), V3 = c("y",2,3), V4 = c("z",4,5))

DF2 = data.frame(V1 = c("","A", "B"), V2 = c("x",6,7), V3 = c("y",8,9), V4 = c("z",0,0))

两个数据框的原始输出如下：

> DF1                > DF2
   V1 V2 V3 V4          V1 V2 V3 V4
1     x  y  z        1     x  y  z
2  A  0  2  4        2  A  6  8  0
3  B  1  3  5        3  B  7  9  0

需求是将数据框的第一行设为列名，调整后效果如下：

>DF1                > DF2
    x y z               x y z
1 A 0 2 4           1 A 6 8 0
2 B 1 3 5           2 B 7 9 0

当前已实现的功能代码如下：

if("V2" %in% names(DF1)){
  names(DF1) = as.character(unlist(DF1[1,]))
  DF1 = DF1[-1, ]
}
if("V2" %in% names(DF2)){
  names(DF2) = as.character(unlist(DF2[1,]))
  DF2 = DF2[-1, ]
}

上述代码可正常实现需求。

问题 #

如何避免编写两个逻辑完全相同的if语句？我首先想到的是用循环遍历两个DF，但尝试后发现由于需要重命名DF对象无法生效；更泛化的问题是，当循环无法使用时，如何避免对多个参数执行重复的相同操作？

核心思路是把重复逻辑抽成公共函数，再批量处理多个数据框即可，有两种常用实现方式：

方法1：封装自定义函数，批量处理后赋值回原对象 #

首先把重复的逻辑写成独立的处理函数：

# 定义统一的列名调整函数
set_first_row_as_colnames <- function(df) {
  if ("V2" %in% names(df)) {
    names(df) <- as.character(unlist(df[1, ]))
    df <- df[-1, ]
  }
  # 可按需重置行号，避免处理后行号从2开始
  rownames(df) <- NULL
  return(df)
}

如果要直接修改原环境里的DF1、DF2，可以用lapply+assign批量赋值：

# 把要处理的数据框名存成字符向量
df_names <- c("DF1", "DF2")
# 批量处理并赋值回原对象
invisible(lapply(df_names, function(name) {
  assign(name, set_first_row_as_colnames(get(name)), envir = .GlobalEnv)
}))

方法2：把数据框放进列表统一管理（更符合R的代码风格） #

如果你的数据框是同类型的批量数据，建议直接存在列表里，不用单独定义多个变量，后续所有批量操作都会更方便：

# 把所有数据框存进列表
df_list <- list(DF1 = DF1, DF2 = DF2)
# 批量处理所有元素
df_list <- lapply(df_list, set_first_row_as_colnames)

处理完之后可以通过df_list$DF1、df_list$DF2调用对应的数据框，也可以再单独赋值回原环境变量。

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