了解 coxph 模型的 R survfit.plot 函数中的颜色

如何解决了解 coxph 模型的 R survfit.plot 函数中的颜色

我正在尝试创建一组带有分类解释变量的单变量 coxph 模型，对于每个模型，我想按变量级别比较生存函数。例如我正在创建：

library(survival)

set.seed(4321)
data <- data.frame(cbind(
  sample(0:1,40,replace=TRUE),# event
  sample(0:24,# time
  sample(1:2,# Diam
  sample(3:4,replace=TRUE) # N
  ))

colnames(data) <- c("event","time","Diam","N")

data$Diam <- as.factor(data$Diam)
data$N <- as.factor(data$N)

model1 <- coxph(Surv(time,event) ~ Diam,data = data)
model2 <- coxph(Surv(time,event) ~ N,data = data)

其中 Diam 和 N 有两个级别。我准备了这段代码来绘制生存函数：

plot(survfit(model1,newdata = data),fun = "s",conf.int = TRUE,col = 2:4,xlab = "Months",ylab = "Proportion without event",main = "Survival curve by Diam")
legend("bottomleft",legend = levels(data$Diam),lty = 1,col = c(2,4))

plot(survfit(model2,main = "Survival curve by N")
legend("bottomleft",legend = levels(data$N),4))

这是输出：

First variable

Second variable

我不明白：

为什么第二个图表中的颜色不同？
如何在图表和图例中匹配颜色。

任何帮助将不胜感激！先感谢您，弗朗西斯卡