编写R函数绘制Kaplan-Meier曲线遇对象未找到错误，求解决方案

问题概述 #

有一个用于生存分析的数据集my_data，包含：

• time：事件发生时间；
• status：0/1编码的事件状态；
• 多个影响因子（factor1、factor2等）。

使用survival和survminer包的基础代码可正常运行：

surv_object1 <- survfit(Surv(time, status) ~ factor1, data = my_data)
ggsurvplot(surv_object1)

surv_object2 <- survfit(Surv(time, status) ~ factor2, data = my_data)
ggsurvplot(surv_object2)

为简化代码编写了如下函数，但调用survival_plot(factor1)时报错：Error in eval(predvars, data, env) : object 'factor1' not found

survival_plot <- function(factor) {
    
    survival_object <- survfit(Surv(time, status) ~ factor, data = my_data)
    surv_plot <- ggsurvplot(survival_object)
    
    return(surv_plot)
}

最小可复现示例：

library(survival)
library(survminer)

survival_plot <- function(factor) {
    
    s_obj <- survfit(Surv(rfstime, status) ~ factor, data = survival::gbsg)
    surv_plot <- ggsurvplot(s_obj)
    
    return(surv_plot)
}

survival_plot(meno)

错误原因 #

函数报错的核心是公式的静态解析问题：

• 函数内部的公式Surv(rfstime, status) ~ factor中，factor被当作字面量变量名，而非传入的参数值（如meno）；
• 调用survival_plot(meno)时，meno会在全局环境中被解析，但函数内部的公式无法关联到这个传入的变量，导致R在数据集里寻找名为factor的列，最终报错。

解决方案 #

以下是几种可行的修复方案，解决动态公式的解析问题：

方案1：动态构建公式（字符串拼接） #

直接将传入的因子名转为字符串，拼接成合法的生存分析公式：

library(survival)
library(survminer)

survival_plot <- function(factor_name) {
    # 拼接公式字符串并转为formula对象
    surv_formula <- as.formula(paste("Surv(rfstime, status) ~", factor_name))
    
    s_obj <- survfit(surv_formula, data = survival::gbsg)
    return(ggsurvplot(s_obj))
}

# 调用时传入变量名的字符串
survival_plot("meno")

方案2：使用rlang准引用（直接传入变量名） #

如果希望直接传入变量名而非字符串，可借助rlang包的准引用工具，确保变量被正确映射到数据集列：

library(survival)
library(survminer)
library(rlang)

survival_plot <- function(factor) {
    # 捕获传入的变量符号
    factor_sym <- enquo(factor)
    
    s_obj <- survfit(Surv(rfstime, status) ~ !!factor_sym, data = survival::gbsg)
    return(ggsurvplot(s_obj))
}

# 直接传入变量名调用
survival_plot(meno)

方案3：优化参考方案 #

对参考的解决方案进行简化，无需额外创建Surv列：

library(survival)
library(survminer)

survival_function <- function(strata_x) {
    survFormula <- as.formula(paste("Surv(rfstime, status) ~", strata_x))
    my_survfit <- survfit(survFormula, data = survival::gbsg)
    # 修正survfit对象的call属性，避免ggsurvplot解析异常
    my_survfit$call$formula <- survFormula
    return(ggsurvplot(my_survfit, data = survival::gbsg))
}

# 传入变量名字符串调用
survival_function("meno")

关键提示 #

• 若传入字符串形式的变量名，as.formula()拼接是最直接的实现方式；
• 若要直接传入变量名符号，需用rlang的enquo()和!!完成准引用，确保R正确识别变量对应的数据列；
• 部分场景下ggsurvplot会读取survfit对象的call属性，修正my_survfit$call$formula可避免绘图时的解析错误。

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