Overlevingscurves plotten en vergelijken
Terug in de hartkliniek wil je aan het hoofd Onderzoek visualisaties laten zien die de verschillen in overlevingstijden van patiënten duidelijk maken. Je plot de overlevingscurves van patiënten met en zonder pericardiale effusie naast elkaar.
Een KaplanMeierFitter-object kmf is al geïnstantieerd. DataFrames voor patiënten met en zonder pericardiale effusie zijn geladen en opgeslagen als respectievelijk has_pericardial_effusion en none_pericardial_effusion.
De klasse KaplanMeierFitter is voor je geïmporteerd. Daarnaast is het pakket pandas geladen als pd, en de module matplotlib.pyplot als plt.
Deze oefening maakt deel uit van de cursus
Survivalanalyse in Python
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Fit kmf to patients with pericardial effusion
____.____(____, ____, label='has_pericardial_effusion')
# Create a plot of the survival function
surv_plot = kmf.____()