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

Question

我不知道如何修改lifelines提供的输出图像，因为我不熟悉"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)