Teken een overlevingscurve
Je bent een socioloog die onderzoekt hoe ex-gedetineerden zich gedragen na hun vrijlating. De DataFrame prison bevat gegevens van 432 veroordeelden die in de jaren 70 uit gevangenissen in de staat Maryland zijn vrijgelaten en één jaar na vrijlating zijn gevolgd. Er zijn twee hoofdkolommen van belang:
week: het aantal weken sinds vrijlatingarrest: of de veroordeelde binnen één jaar na vrijlating opnieuw is gearresteerd
Er is al een Kaplan-Meier-overlevingsfunctie voor je gefit en het modelobject heet prison_kmf. Kun je de overlevingscurve van prison_kmf visualiseren?
De bibliotheek matplotlib.pyplot is geïmporteerd als plt en de module lifelines is ook voor je geïmporteerd.
Deze oefening maakt deel uit van de cursus
Survivalanalyse in Python
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Plot the survival function
____
# Show the plot
____