ÎncepețiÎncepe gratuit

Trasează o curbă de supraviețuire

Ești un sociolog care studiază comportamentul foștilor deținuți după eliberare. DataFrame-ul prison conține informații despre 432 de persoane eliberate din închisorile statului Maryland în anii '70, urmărite timp de un an după eliberare. Există două coloane principale de interes:

  • week: numărul de săptămâni scurse de la eliberare
  • arrest: dacă persoana a fost reîncarcerată în decurs de un an după eliberare

O funcție de supraviețuire Kaplan-Meier a fost ajustată pentru tine, iar obiectul modelului se numește prison_kmf. Poți vizualiza curba de supraviețuire a lui prison_kmf?

Biblioteca matplotlib.pyplot este importată ca plt, iar modulul lifelines a fost de asemenea importat pentru tine.

Acest exercițiu face parte din cursul

Analiza de supraviețuire în Python

Vezi cursul

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Plot the survival function
____

# Show the plot
____
Editează și rulează codul