lifelines_kaplan_meier_survival_curve_fit_and_plot.py

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