R语言：如何传递列名向量实现mtcars数据集分组均值汇总

1. 正确传递列名向量到函数参数 #

原代码的核心问题是字符串列名无法被统计函数直接识别，需要用tidyverse的语法处理字符串到列引用的转换，以下是两种可行方案：

方案1：使用.data代词（直观易读） #

dplyr提供的.data代词可以直接通过字符串定位数据框列：

library(dplyr)

mtcars.vars = c("mpg", "cyl", "disp")
summary_function <- function(statistic){
  # 初始化空列表存储每个变量的结果
  result_list <- list()
  for(col in mtcars.vars){
    var_summary <- mtcars %>%
      group_by(am) %>%
      summarise(mean = statistic(.data[[col]], na.rm = TRUE))
    # 用列名作为列表元素的名称
    result_list[[col]] <- var_summary
  }
  return(result_list)
}

# 调用函数获取所有结果
all_mean_results <- summary_function(mean)

方案2：使用符号转换（非标准求值） #

将字符串列名转为符号，再通过!!强制求值：

library(dplyr)
library(rlang)

mtcars.vars = c("mpg", "cyl", "disp")
summary_function <- function(statistic){
  result_list <- list()
  for(col in mtcars.vars){
    col_sym <- sym(col)
    var_summary <- mtcars %>%
      group_by(am) %>%
      summarise(mean = statistic(!!col_sym, na.rm = TRUE))
    result_list[[col]] <- var_summary
  }
  return(result_list)
}

all_mean_results <- summary_function(mean)

2. 保存带列名后缀的独立变量 #

如果不想用列表存储，而是生成如mpg_mean、cyl_mean这类带后缀的独立变量，可以用assign()函数在循环中创建：

library(dplyr)

mtcars.vars = c("mpg", "cyl", "disp")
summary_function <- function(statistic){
  for(col in mtcars.vars){
    var_summary <- mtcars %>%
      group_by(am) %>%
      summarise(mean = statistic(.data[[col]], na.rm = TRUE))
    # 生成带后缀的变量名
    var_name <- paste0(col, "_mean")
    # 将结果写入全局环境
    assign(var_name, var_summary, envir = .GlobalEnv)
  }
}

# 调用后，全局环境会出现mpg_mean、cyl_mean、disp_mean三个变量
summary_function(mean)

补充：无循环的简洁写法 #

用purrr包的map函数可以替代循环，代码更简洁：

library(dplyr)
library(purrr)

mtcars.vars = c("mpg", "cyl", "disp")
all_mean_results <- mtcars.vars %>%
  map(~ mtcars %>%
        group_by(am) %>%
        summarise(mean = mean(.data[[.x]], na.rm = TRUE))) %>%
  set_names(mtcars.vars)

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