有没有一种方法可以从CPH fitter软件包中的plot_协变量_组方法中提取数据？

Question

我使用生命线中的CPH.fitter运行了一个Cox比例风险模型。现在我想要队列中一些个体参数的生存函数。使用plot_协变量_群方法，我得到了生存函数的图像，但我需要提取生成该图的数据。有人能帮我吗

提前谢谢你

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)