R语言中如何让自定义t.test函数支持传入字符型列名而非数值向量

问题核心 #

你自定义的t_test函数默认接收数值向量作为输入，当传入列名字符串时，函数无法将字符串解析为数据框中对应的列，导致公式var ~ gr无法被t.test识别。

解决方法 #

1. 改造自定义函数，支持传入列名字符串

修改函数逻辑，将传入的列名字符串转为公式对象，同时将数据框作为参数传入，避免全局环境变量冲突：

t_test <- function(var_name, df = data){
  # 拼接公式字符串并转为公式对象
  test_formula <- as.formula(paste0(var_name, " ~ gr"))
  test_res <- t.test(test_formula, data = df)
  
  p_val <- test_res$p.value
  t_stat <- test_res$statistic
  des <- ifelse(p_val < 0.05, "good", "bad")
  
  return(list(p_value = p_val, t_stat = t_stat, decision = des))
}

改造完成后，传入带引号的列名即可正常运行：

t_test('x')

2. 优化批量计算逻辑

你原来的循环可以直接复用改造后的函数，也可以用lapply简化写法，避免手动初始化空数据框的繁琐操作：

# 提取需要计算的列名
data_colnames <- names(data)[2:ncol(data)]
# 批量计算并合并结果
t_test_data <- do.call(rbind, lapply(data_colnames, function(col) {
  res <- t_test(col)
  data.frame(
    variable = col,
    t_stat = res$t_stat,
    p_value = res$p_value,
    decision = res$decision,
    check.names = FALSE
  )
}))

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