R语言导出带文本表头的Excel：多数据集与多输出处理问题

一、现有报错修复 #

1. 报错 Error in if (file.exists(file) & !overwrite) { : the condition has length > 1 #

问题出在第二个lapply调用：lapply(output_data, name_file, names(res))会把整个文件名向量传给每个name_file实例，导致paste0(file_name, '.xlsx')生成多个文件名，而saveWorkbook仅支持单个文件名输入，触发条件判断长度异常。

修复方式：改用Map（或mapply）实现数据集与文件名的一一对应遍历：

Map(name_file, output_data, names(res))

2. 第二个lapply返回全0列表 #

这是正常现象：saveWorkbook执行成功时会返回invisible(0)，lapply会收集这些返回值形成列表。改用Map后同样会返回执行状态，但能保证每个文件正确生成。

二、处理函数返回双结果的实现 #

1. 修改处理函数返回命名列表 #

原函数未明确返回值，需调整为返回包含dat1和dat2的命名列表，同时修正group_by的非标准求值问题（原代码用'vars'会把字符串当作分组列）：

some_fun = function(dat, vars){
  dat1 <- dat %>%
    group_by({{vars}}) %>%  # 使用{{}}引用传入的变量参数
    summarize(count = n())
  
  dat2 <- dat %>%
    group_by({{vars}}) %>%
    mutate(count = n())
  
  list(raw = dat1, processed = dat2)  # 返回命名列表，方便后续调用
}

2. 修改导出函数适配双结果 #

调整函数接收列表输入，分别写入对应工作表，并添加指定的两行标题：

name_file = function(dat_list, file_name){
  wb <- createWorkbook()
  
  # 写入raw工作表，添加两行标题
  addWorksheet(wb, 'raw')
  writeData(wb, 'raw', "Report", startCol = 1, startRow = 1)
  writeData(wb, 'raw', "Quarter", startCol = 1, startRow = 2)
  writeData(wb, 'raw', dat_list$raw, startCol = 1, startRow = 3)
  
  # 写入processed工作表，添加两行标题
  addWorksheet(wb, 'processed')
  writeData(wb, 'processed', "Report", startCol = 1, startRow = 1)
  writeData(wb, 'processed', "Quarter", startCol = 1, startRow = 2)
  writeData(wb, 'processed', dat_list$processed, startCol = 1, startRow = 3)
  
  saveWorkbook(wb, file = paste0(file_name, '.xlsx'), overwrite = TRUE)
}

3. 完整可运行代码 #

library(openxlsx)
library(dplyr)

# 构造示例数据集列表
res <- list(mtcars, iris)
names(res) <- c('mtcars', 'iris')

# 双返回值处理函数
some_fun = function(dat, vars){
  dat1 <- dat %>%
    group_by({{vars}}) %>%
    summarize(count = n())
  
  dat2 <- dat %>%
    group_by({{vars}}) %>%
    mutate(count = n())
  
  list(raw = dat1, processed = dat2)
}

# Excel导出函数
name_file = function(dat_list, file_name){
  wb <- createWorkbook()
  
  addWorksheet(wb, 'raw')
  writeData(wb, 'raw', "Report", startCol = 1, startRow = 1)
  writeData(wb, 'raw', "Quarter", startCol = 1, startRow = 2)
  writeData(wb, 'raw', dat_list$raw, startCol = 1, startRow = 3)
  
  addWorksheet(wb, 'processed')
  writeData(wb, 'processed', "Report", startCol = 1, startRow = 1)
  writeData(wb, 'processed', "Quarter", startCol = 1, startRow = 2)
  writeData(wb, 'processed', dat_list$processed, startCol = 1, startRow = 3)
  
  saveWorkbook(wb, file = paste0(file_name, '.xlsx'), overwrite = TRUE)
}

# 执行处理与导出
output_data = lapply(res, some_fun, vars = cyl)  # mtcars用cyl分组，iris可替换为Species
Map(name_file, output_data, names(res))

三、自定义文件名子集提取 #

若需从原文件名中提取子集作为导出文件名，可使用字符串处理函数提前处理，示例：

# 假设原名为"dataset_2024Q1"，提取季度部分
library(stringr)
new_file_names <- str_extract(names(res), "\\d{4}Q\\d")
Map(name_file, output_data, new_file_names)

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