如何修改cph.plot_covariate_groups绘制的COXPH图像的输出

Question

我不知道如何修改生命线提供的输出图像，因为我不熟悉“ cph.plot_covariate_groups”。不幸的是，这里的链接中似乎没有关于它的详细描述。 https://lifelines.readthedocs.io/en/latest/Survival%20Regression.html。

我要寻找的是，（1）如何缩短事件天数（X轴），我不想显示如此长的生存曲线。理想情况下，最好是4000。 （2）另外，如果可能，我想从图像中删除基线生存曲线。 （3）我也希望能将生存曲线的颜色从橙色/蓝色更改为其他颜色。

import pandas as pd
from lifelines import AalenAdditiveFitter, CoxPHFitter, KaplanMeierFitter

data =  pd.read_csv("cluster label.csv", index_col=0)
cph = CoxPHFitter()
cph.fit(data, duration_col="time", event_col="status")
cph.plot_covariate_groups('label', [0,1])

Answer 1

这一切皆有可能。有关特定功能和方法的信息可在文档页面上找到：https://lifelines.readthedocs.io/en/latest/References.html

具体来说：https://lifelines.readthedocs.io/en/latest/lifelines.fitters.html#lifelines.fitters.coxph_fitter.CoxPHFitter.plot_covariate_groups

所以尝试一下：

cph.plot_covariate_groups('label', [0,1], 
   plot_baseline=False, 
   cmap='coolwarm'
)
plt.xlim(0, 4000)