Analyse de survie en Python
Shae Wang
Senior Data Scientist
Données jouets avec $n=5$ :
| durée | observé |
|---|---|
| 2 | 1 |
| 5 | 0 |
| 3 | 1 |
| 5 | 1 |
| 2 | 0 |
Étape 1 : Ordonner les données par ordre croissant. En cas d'égalité, placer les données censurées après les non censurées.
Étape 2 : Pour chaque $t_i$, calculer $d_i$, $n_i$ et $\big(1-\frac{d_i}{n_i}\big)$
Étape 3 : Pour chaque $t_i$, multiplier $\big(1-\frac{d_i}{n_i}\big)$ par $\big(1-\frac{d_{i-1}}{n_{i-1}}\big)$, $\big(1-\frac{d_{i-2}}{n_{i-2}}\big)$, … , $\big(1-\frac{d_0}{n_0}\big)$
Étape 1 : Ordonner les durées par ordre croissant. En cas d'égalité, placer les données censurées après les non censurées.
| durée |
|---|
| 2 |
| 5+ |
| 3 |
| 5 |
| 2+ |
Utilisez le signe « + » pour indiquer les données censurées : 2, 5+, 3, 5, 2+
Étape 1 : Ordonner les durées par ordre croissant. En cas d'égalité, placer les données censurées après les non censurées.
| $t_i$ |
|---|
| 2, 2+ |
| 3 |
| 5, 5+ |
Étape 2 : Pour chaque $t_i$, calculer $d_i$, $n_i$ et $\big(1-\frac{d_i}{n_i}\big)$
| $t_i$ |
|---|
| 2, 2+ |
| 3 |
| 5, 5+ |
Étape 2 : Pour chaque $t_i$, calculer $d_i$, $n_i$ et $\big(1-\frac{d_i}{n_i}\big)$
| $t_i$ | $d_i$ |
|---|---|
| 2, 2+ | 1 |
| 3 | 1 |
| 5, 5+ | 1 |
Étape 2 : Pour chaque $t_i$, calculer $d_i$, $n_i$ et $\big(1-\frac{d_i}{n_i}\big)$
| $t_i$ | $d_i$ | $n_i$ |
|---|---|---|
| 2, 2+ | 1 | 5 |
| 3 | 1 | 3 |
| 5, 5+ | 1 | 2 |
Étape 2 : Pour chaque $t_i$, calculer $d_i$, $n_i$ et $\big(1-\frac{d_i}{n_i}\big)$
| $t_i$ | $d_i$ | $n_i$ | $\big(1-\frac{d_i}{n_i}\big)$ |
|---|---|---|---|
| 2, 2+ | 1 | 5 | $4/5$ |
| 3 | 1 | 3 | $2/3$ |
| 5, 5+ | 1 | 2 | $1/2$ |
Étape 3 : Pour chaque $t_i$, multiplier $\big(1-\frac{d_i}{n_i}\big)$ par $\big(1-\frac{d_{i-1}}{n_{i-1}}\big)$, $\big(1-\frac{d_{i-2}}{n_{i-2}}\big)$, … , $\big(1-\frac{d_0}{n_0}\big)$
| $t_i$ | $d_i$ | $n_i$ | $\big(1-\frac{d_i}{n_i}\big)$ | $S(t_i)$ |
|---|---|---|---|---|
| 2, 2+ | 1 | 5 | 4/5 | 4/5 = 0,8 |
| 3 | 1 | 3 | 2/3 | 4/5 $\cdot$ 2/3 = 0,53 |
| 5, 5+ | 1 | 2 | 1/2 | 4/5 $\cdot$ 2/3 $\cdot$ 1/2 = 0,27 |
| $t_i$ | $d_i$ | $n_i$ | $\big(1-\frac{d_i}{n_i}\big)$ | $S(t_i)$ |
|---|---|---|---|---|
| 2, 2+ | 1 | 5 | $4/5$ | 0,8 |
| 3 | 1 | 3 | $2/3$ | 0,53 |
| 5, 5+ | 1 | 2 | $1/2$ | 0,27 |


Les probabilités de survie à chaque instant entre 0 et 5.
Idée reçue : si la courbe atteint 0, aucun sujet n'a survécu.
from lifelines import KaplanMeierFitter
import matplotlib.pyplot as plt
kmf = KaplanMeierFitter()
kmf.fit(durations, event_observed)
kmf.survival_function_.plot()
plt.show()
Nom du DataFrame : mortgage_df
| id | durée | remboursé |
|---|---|---|
| 1 | 25 | 0 |
| 2 | 17 | 1 |
| 3 | 5 | 0 |
| ... | ... | ... |
| 100 | 30 | 1 |
from lifelines import KaplanMeierFitter
from matplotlib import pyplot as plt
mortgage_kmf = KaplanMeierFitter()
mortgage_kmf.fit(duration=mortgage_df["duration"],
event_observed=mortgage_df["paid_off"])
mortgage_kmf.survival_function_.plot()
plt.show()

mortgage_kmf.plot_survival_function()
plt.show()

Tracer les estimations ponctuelles de la fonction de survie en ligne continue.
kmf.survival_function_.plot()
plt.show()

Tracer la fonction de survie en ligne en escalier sans intervalle de confiance.
kmf.plot(ci_show=False)
plt.show()

Tracer la fonction de survie en ligne en escalier avec l'intervalle de confiance.
kmf.plot()
plt.show()

Autre option…
kmf.plot_survival_function()
plt.show()
Analyse de survie en Python