Rysowanie krzywej przeżycia

Jesteś socjologiem badającym zachowanie skazanych po zwolnieniu z więzienia. DataFrame prison zawiera dane dotyczące 432 skazanych zwolnionych z zakładów karnych stanu Maryland w latach 70. XX wieku, których losy śledzono przez rok po zwolnieniu. Dwie główne kolumny to:

• week: liczba tygodni, które upłynęły od zwolnienia
• arrest: czy skazany został ponownie aresztowany w ciągu roku od zwolnienia

Funkcja przeżycia Kaplana-Meiera została już dla ciebie dopasowana, a obiekt modelu nosi nazwę prison_kmf. Czy potrafisz zwizualizować krzywą przeżycia prison_kmf?

Biblioteka matplotlib.pyplot jest zaimportowana jako plt, a moduł lifelines również został zaimportowany.