R在一个图中绘制多个模型

Question

我已经使用 kaplan-Meir 方法完成了多个生存模型，每个生存模型都是通过根据数据表中显示的组列将数据子集提取到不同的 R 数据表来构建的。我可以单独绘制每条生存曲线，但我想在一个图中绘制所有这些不同的模型。最好的方法是什么。

 userid        lifespan_days              event        group
2                    4657                    1           A
4                    4658                    1           A
16                   1106                    1           A
50                    458                    1           A
51                   4393                    1           A
57                    305                    1           A

在 ggplot 中执行此操作会很棒，通过搜索我发现关注 ggplot2 - plot multiple models on the same plot，但由于我的数据性质，我在执行此类场景时遇到问题。例如userid 来自多个网站，因此userid=2 可以存在于另一个组下。

假设使用上面的 data.table 我创建了以下内容：

a_time <- dt$lifespan_days
a_event <- dt$event
survival_model_a <- survfit(Surv(a_time, a_event) ~ 1)
plot(survival_model_a)

这只会在同一个图中绘制一个相似之处

Answer 1

您可以使用以下方法绘制所有模型，而无需子集：

dt <- read.table(header=T, text="userid        lifespan_days              event        group
2                    4657                    1           A
4                    4658                    0           A
16                   1106                    1           B
50                    458                    1           B
51                   4393                    1           C
57                    305                    1           A")

library(survival)
a_time <- dt$lifespan_days
a_event <- dt$event
survival_model_a <- survfit(Surv(a_time, a_event) ~ dt$group)
plot(survival_model_a, col = rainbow(length(unique(dt$group))))

Answer 2

使用lines() 进行后续的绘图调用，如下所示：

b_time <- dtB$lifespan_days
b_event <- dtB$event
survival_model_b <- survfit(Surv(b_time, b_event) ~ 1)
lines(survival_model_b)

如果你想使用ggplot2，this question有两个很好的答案。