1. Nauka
    2. /
  2. Kursy
    3. /
  3. Analiza przeżycia w Pythonie
Connected

ćwiczenie

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.

Instrukcje 1/4

undefined XP
    1
    2
    3
    4
  • Utwórz maskę boolowską dla pacjentów, którzy otrzymali nowe leczenie.