如何在R语言表格中添加通用序列号和表头，封装函数时序列号默认、表头可参数化

需求说明 #

• 自动生成序列号列，无需手动赋值
• 表头内容支持作为参数传入，方便后续封装为复用函数
• 可兼容原有数据框处理逻辑

实现代码 #

基础操作示例 #

library(tibble)
library(dplyr)

# 基于mtcars数据集的基础处理逻辑
df <- mtcars %>%
  rownames_to_column(var = "carnames") %>%
  select(1:2) %>%
  as.data.frame()

# 传入自定义表头
custom_headers <- c("车辆名称", "气缸数")
colnames(df) <- custom_headers

# 自动生成序列号列并置于第一列
df <- df %>%
  mutate(序列号 = row_number(), .before = 1)

封装后可复用函数 #

process_df <- function(input_df, custom_headers, serial_col_name = "序列号") {
  # 校验表头长度与数据框列数是否匹配
  if(length(custom_headers) != ncol(input_df)) {
    stop("自定义表头数量与数据框列数不匹配，请检查输入")
  }
  # 应用自定义表头
  colnames(input_df) <- custom_headers
  # 生成序列号列
  output_df <- input_df %>%
    mutate(!!sym(serial_col_name) := row_number(), .before = 1)
  return(output_df)
}

# 函数调用示例
df_raw <- mtcars %>%
  rownames_to_column(var = "carnames") %>%
  select(1:2) %>%
  as.data.frame()

# 传入自定义表头获取处理结果
result_df <- process_df(df_raw, custom_headers = c("车辆名称", "气缸数"))

函数参数说明 #

• input_df：待处理的输入数据框
• custom_headers：自定义表头向量，长度需与输入数据框的列数一致
• serial_col_name：序列号列的列名，默认值为「序列号」，可按需修改

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