Niech żyje demokracja!

Wcześniej w tym rozdziale zbadaliśmy i potwierdziliśmy problem cenzurowania danych w zbiorze dotyczącym czasu trwania reżimów. Teraz czas zbudować pierwszą krzywą przeżycia do szacowania średniego czasu trwania.

Twoim zadaniem jest oszacowanie funkcji przeżycia i narysowanie krzywej przeżycia przedstawiającej prawdopodobieństwa przeżycia. Aby oszacować funkcję przeżycia, utwórz instancję obiektu KaplanMeierFitter i dopasuj go do odpowiednich kolumn danych. Nazwy parametrów zostały już podane, ponieważ więcej o dopasowywaniu funkcji przeżycia dowiesz się w kolejnych rozdziałach.

Pakiet pandas jest załadowany jako pd, moduł matplotlib.pyplot jako plt, a klasa KaplanMeierFitter jest zaimportowana z lifelines. Ramka danych z czasami trwania jest załadowana i przechowywana jako regime_durations.