Porównanie metod leczenia bólu u pacjentów

Trwa eksperyment dotyczący nowej metody leczenia bólu. Chcesz porównać pacjentów, którzy otrzymali nowe leczenie, z tymi, którzy otrzymali stare. Dopasuj estymator Kaplana-Meiera do każdego zbioru danych i zwizualizuj krzywe przeżycia obok siebie.

DataFrame o nazwie recur zawiera następujące kolumny:

• age: wiek pacjenta;
• treat: zastosowane leczenie (0 – nowe, 1 – stare);
• duration: czas po leczeniu w godzinach;
• censor: informacja o tym, czy zdarzenie jest cenzurowane;

Pakiet pandas jest załadowany jako pd, klasa KaplanMeierFitter została zaimportowana z lifelines, a moduł pyplot z matplotlib – jako plt.

Dodatkowo zostały już utworzone: instancja KaplanMeierFitter o nazwie kmf oraz wykres z podwykresem o nazwie ax.