lifelines_kaplan_meier_survival_curve_quickstart.py

1from lifelines import KaplanMeierFitter
2from lifelines.datasets import load_waltons
3import matplotlib.pyplot as plt
4
5# Load the waltons dataset
6df = load_waltons()
7
8# T is the duration, E is the event observed (1 if died, 0 if censored)
9T = df['T']
10E = df['E']
11
12# Initialize the KaplanMeierFitter
13kmf = KaplanMeierFitter()
14
15# Fit the data
16kmf.fit(T, event_observed=E)
17
18# Plot the survival curve
19kmf.plot_survival_function()
20plt.title('Survival function of Walton\'s mice')
21plt.show()