lifelines_kaplan_meier_survival_curve_lung_cancer_plot.py

1from lifelines import KaplanMeierFitter
2from lifelines.datasets import load_lung
3import matplotlib.pyplot as plt
4
5# Load the example dataset
6data = load_lung()
7
8# T is the duration, E is the binary event observed (1 if death, 0 if censored)
9T = data['time']
10E = data['status']
11
12# Initialize and fit the model
13kmf = KaplanMeierFitter()
14kmf.fit(T, event_observed=E)
15
16# Plot the survival function
17kmf.plot_survival_function()
18plt.title('Survival function of lung cancer patients')
19plt.show()