在R中循环多数据框生成按客户拆分的多标签Excel文件

我有多个数据框，每个数据框需通过openxlsx包填充Excel的指定标签。每个数据框中包含多个客户数据，希望自动化生成文件：遍历客户列表，将对应数据写入预定义标签，完成一个客户的所有标签后，以客户名命名并保存Excel文件，再处理下一个客户。

示例数据如下：

library(tidyverse)
library(openxlsx)

df.1 <- tribble(
  ~customer  ,~period, ~cost1, ~cost2 ,
  'cust1',  '202201', 5, 10,
  'cust1',  '202202', 5, 10,
  'cust1',  '202203', 5, 10,
  'cust1',  '202204', 5, 10,
  'cust2',  '202203', 5, 10,
  'cust2',  '202204', 5, 10,
  'cust2',  '202202', 5, 10,
  'cust3',  '202204', 5, 10,
  
)

df.2 <- tribble(
  ~customer  ,~period, ~cost3,
  'cust1',  '202201', 5,
  'cust1',  '202202', 5,
  'cust1',  '202203', 5,
  'cust1',  '202204', 5,
  'cust2',  '202203', 5,
  'cust3',  '202203', 5,
  'cust3',  '202204', 5,
  'cust4',  '202201', 5,
)

df.1_cust <- df.1 %>% select(customer) %>% distinct()
df.2_cust <- df.2 %>% select(customer) %>% distinct()

cust_list <- df.1_cust %>% 
  rbind(df.2_cust) %>% 
  distinct()

我的循环尝试代码如下：

tab1_data <- df.1 # the data that will go into tab 1
tab2_data <- df.2  # the data that will go into tab 2


for (i in 1:length(cust_list)) {
wb <- openxlsx::createWorkbook()
openxlsx::addWorksheet(wb, 'tab1')
openxlsx::addWorksheet(wb, 'tab2')

openxlsx::writeData(tab1, startCol = 1, startRow = 1,x = tab1_data[i])
openxlsx::writeData(tab2, startCol = 1, startRow = 1,x = tab2_data[i])

openxlsx::saveWorkbook(wb, overwrite = T)
}

预期输出为4个独立的Excel文件（每个客户一个），每个文件包含tab1和tab2两个标签，文件名与客户名对应。请问如何实现该需求？

你的现有代码存在几个关键问题，修正后即可实现需求：

问题点分析 #

1. 循环索引错误：cust_list是数据框，length(cust_list)返回列数（此处为1），无法遍历所有客户，需提取客户名称向量循环。
2. 数据筛选错误：tab1_data[i]仅取数据框第i列，未按客户筛选数据，需用filter提取当前客户的行。
3. writeData参数错误：第一个参数应为工作簿对象wb，需指定工作表名称，而非直接写tab1。
4. 保存路径缺失：saveWorkbook需指定文件名，否则会报错或覆盖同一文件。

修正后的代码 #

# 提取客户名称向量，简化循环逻辑
cust_names <- pull(cust_list, customer)

for (cust in cust_names) {
  # 创建新工作簿
  wb <- createWorkbook()
  
  # 添加指定工作表
  addWorksheet(wb, "tab1")
  addWorksheet(wb, "tab2")
  
  # 筛选当前客户的专属数据
  tab1_filtered <- df.1 %>% filter(customer == cust)
  tab2_filtered <- df.2 %>% filter(customer == cust)
  
  # 将数据写入对应工作表
  writeData(wb, sheet = "tab1", x = tab1_filtered, startCol = 1, startRow = 1)
  writeData(wb, sheet = "tab2", x = tab2_filtered, startCol = 1, startRow = 1)
  
  # 按客户名命名并保存文件
  saveWorkbook(wb, file = paste0(cust, ".xlsx"), overwrite = TRUE)
}

代码说明 #

• 提取客户向量：用pull将cust_list中的客户名称转为字符向量，循环时直接遍历每个客户名，逻辑更清晰。
• 数据筛选：每次循环通过filter筛选当前客户的所有行，确保写入的数据仅属于该客户。
• writeData参数：第一个参数传入工作簿wb，通过sheet指定目标工作表，保证数据写入正确标签。
• 命名保存：用paste0拼接客户名与.xlsx后缀，每个客户生成独立文件，避免覆盖。

运行后将生成cust1.xlsx、cust2.xlsx、cust3.xlsx、cust4.xlsx四个文件，每个文件包含两个标签，对应客户的专属数据。

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