有没有办法从 CPH fitter 包中的 plot_covariate_groups 方法中提取数据？

【问题标题】：Is there a way to extract data from plot_covariate_groups method in CPH fitter package?有没有办法从 CPH fitter 包中的 plot_covariate_groups 方法中提取数据？
【发布时间】：2020-08-28 06:10:33
【问题描述】：

我使用来自生命线的 CPH.fitter 运行了 Cox 比例危险模型。现在我想要队列中某些单个参数的生存函数。使用 plot_covariate_groups 方法，我得到了生存函数的图像，但我需要提取创建该图的数据。谁能帮我解决这个问题。

提前谢谢你！

【问题讨论】：

请加minimal code。

标签： python cox-regression lifelines

【解决方案1】：

参考源代码，这样会有所帮助（使用与函数调用相同的参数）

x_bar = cph._norm_mean.to_frame().T
X = pd.concat([x_bar] * values.shape[0])

if np.array_equal(np.eye(n_covariates), values):
    X.index = ["%s=1" % c for c in covariates]
else:
    X.index = [", ".join("%s=%g" % (c, v) for (c, v) in zip(covariates, row)) for row in values]
for covariate, value in zip(covariates, values.T):
    X[covariate] = value

cph.predict_survival_function(X)

【讨论】：