如何用purrr::map迭代带多参数的函数处理动态数据？

问题背景 #

需要编写可灵活配置的函数，为州-城市数据框中的每个州生成独立的gt表格，并用purrr::map实现批量迭代，但原代码在多参数适配和函数逻辑上存在问题。

原代码核心问题 #

• 函数make_gts中return(gt)错误，应返回构建好的gt表格对象，而非gt函数本身
• def_var参数传递的是df$State（向量），而非列名字符串，导致筛选逻辑失效
• map中使用iterate_list[i]是错误语法，map会自动迭代每个元素，无需手动索引
• 变量名错误：ex_data应为示例数据中的df

修正后的完整实现代码 #

library(gt)
library(tidyverse)

# 修正后的gt表格生成函数
make_gts <- function(df, x, def_col, select_vect, title_text){
  # 按指定列筛选目标州数据
  df_ind <- df %>% filter(.data[[def_col]] == x)
  
  # 选择指定列
  df_ind_clean <- df_ind %>% select(all_of(select_vect)) 
  
  # 构建gt表格并返回
  gt_table <- gt(df_ind_clean) %>% 
    tab_header(title = paste("This is a GT for", title_text, "-", x))
  
  return(gt_table)
}

# 示例数据
df <- structure(list(State = c("California", "California", "California", 
"Texas", "Texas", "Texas", "New Mexico", "New Mexico", "New Mexico"
), City = c("Los Angeles", "San Francisco", "Fresno", "Dallas", 
"Austin", "Frisco", "Albuquerque", "Santa Fe", "Taos"), NumObserved = c(1200000L, 
825000L, 113000L, 240000L, 189000L, 38000L, 56000L, 23000L, 6000L
), Elevation = c(28L, 47L, 235L, 312L, 550L, 128L, 4291L, 3533L, 
7823L)), class = "data.frame", row.names = c(NA, -9L))

# 配置参数
title_text = "Elevation and NumObserved"
select_vect <- c("City", "Elevation", "NumObserved")
def_col <- "State" # 传递列名字符串保证灵活性
iterate_list <- unique(df[[def_col]])

# 使用purrr::map迭代生成gt表格列表
state_gt_list <- iterate_list %>% 
  set_names() %>% 
  purrr::map(~make_gts(df = df,
                       x = .,
                       def_col = def_col,
                       select_vect = select_vect,
                       title_text = title_text))

# 查看某州的表格（示例）
state_gt_list$California

更简便的实现方法 #

用group_split直接按州拆分数据框，省去手动筛选步骤，代码更简洁：

# 简化版：无需单独编写筛选逻辑
state_gt_list_simple <- df %>%
  group_by(State) %>%
  group_split() %>%
  set_names(map(., ~unique(.$State))) %>%
  map(~gt(select(., all_of(select_vect))) %>%
        tab_header(title = paste("This is a GT for", title_text, "-", unique(.$State))))

说明 #

• 修正后的函数通过.data[[def_col]]实现对指定列的引用，保证参数灵活性
• purrr::map中用~匿名函数接收迭代的每个州名，传递给make_gts的x参数
• 两种方法最终都会生成以州名为命名的列表，每个元素对应该州的gt表格

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