Porovnání léčby bolestivosti u pacientů

Probíhá experiment s novou léčbou bolestivosti a ty chceš porovnat pacienty, kteří dostali novou léčbu, s těmi, kteří dostali starou. Na každou skupinu pacientů přizpůsobíš Kaplan-Meierův estimátor a výsledné křivky přežití zobrazíš vedle sebe.

DataFrame se jmenuje recur a obsahuje sloupce:

• age: věk pacienta;
• treat: léčba, kterou pacient dostal (0 = nová, 1 = stará);
• duration: doba po léčbě v hodinách;
• censor: zda je událost cenzorována;

Balíček pandas je načtený jako pd, třída KaplanMeierFitter je importována z lifelines a modul pyplot je importován z matplotlib jako plt.

Instance KaplanMeierFitter s názvem kmf a figure se subplotem nazvaná ax jsou již připraveny.