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

Question

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

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

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)

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

How to modify the output of my COXPH image drawn by cph.plot_covariate_groups

python

cox-regression

lifelines