R语言自定义生存分析函数：对象创建与变量引用引号问题

错误原因 #

你遇到的问题核心是：拼接得到的y和z是字符串，而Surv()需要的是数据框中对应的数值型列，直接传入字符串会被Surv()当成非数值变量处理，从而报错。

三种可行解决方案 #

方案1：动态构建公式字符串（基础R实现）

直接拼接完整的生存分析公式字符串，再转换为公式对象传给survfit()：

km_fun <- function(x) {
  # 拼接完整的公式字符串
  formula_str <- paste0("Surv(fu_time_", x, ", fu_status_", x, ") ~ prophied")
  # 转换为公式对象
  formula_obj <- as.formula(formula_str)
  # 计算生存函数并返回结果
  survfit(formula_obj, data = df_ptm)
}

# 测试调用
km_fun("PJP")

方案2：使用rlang包的符号注入（tidyverse风格）

通过sym()将字符串转换为变量符号，再用!!注入到Surv()中：

library(rlang)

km_fun <- function(x) {
  # 将字符串转为变量符号
  time_var <- sym(paste0("fu_time_", x))
  status_var <- sym(paste0("fu_status_", x))
  # 注入符号到公式中
  survfit(Surv(time = !!time_var, event = !!status_var) ~ prophied, data = df_ptm)
}

# 测试调用
km_fun("PJP")

方案3：直接引用数据框列（基础R实现）

直接通过数据框索引提取目标列传入Surv()：

km_fun <- function(x) {
  time_col <- paste0("fu_time_", x)
  status_col <- paste0("fu_status_", x)
  # 直接从数据框中提取列
  survfit(Surv(time = df_ptm[[time_col]], event = df_ptm[[status_col]]) ~ prophied)
}

# 测试调用
km_fun("PJP")

注意事项 #

• 确保df_ptm在函数的作用域内（比如是全局环境中的数据框）；
• 函数最后一行默认返回结果，也可以显式添加return(km_time)增强可读性。

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