Tracer une courbe de survie

Vous êtes sociologue et vous étudiez le comportement des personnes condamnées après leur libération. Le DataFrame prison contient des informations sur 432 personnes libérées des prisons de l’État du Maryland dans les années 1970 et suivies pendant un an après leur sortie. Deux colonnes principales vous intéressent :

• week : le nombre de semaines écoulées depuis la libération
• arrest : indique si la personne a été de nouveau arrêtée dans l’année suivant sa libération

Une fonction de survie de Kaplan-Meier a été ajustée pour vous et l’objet du modèle s’appelle prison_kmf. Pouvez-vous visualiser la courbe de survie de prison_kmf ?

La bibliothèque matplotlib.pyplot est importée sous le nom plt et le module lifelines a également été importé pour vous.