Disegna una curva di sopravvivenza

Sei un/una sociologo/a che studia il comportamento dei detenuti dopo il rilascio. Il DataFrame prison contiene le informazioni su 432 detenuti rilasciati dalle prigioni statali del Maryland negli anni ’70 e seguiti per un anno dopo il rilascio. Ci sono due colonne principali di interesse:

• week: numero di settimane trascorse dal rilascio
• arrest: indica se il detenuto è stato nuovamente arrestato entro un anno dal rilascio

È stata adattata per te una funzione di sopravvivenza di Kaplan-Meier e l’oggetto del modello si chiama prison_kmf. Sai visualizzare la curva di sopravvivenza di prison_kmf?

La libreria matplotlib.pyplot è importata come plt e il modulo lifelines è già stato importato per te.