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

Question

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

先感谢您！

Answer 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)