Есть ли способ извлечь данные из метода plot_covariate_groups в пакете установщика CPH?

Question

Я запустил модель пропорциональной опасности Кокса, используя CPH.fitter из компании lifelines. Теперь мне нужна функция выживания для некоторых индивидуальных параметров в когорте. Используя метод 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)