Порівняння лікування болючості у пацієнтів

Проводиться експеримент із новим методом лікування болючості, і ви хочете порівняти пацієнтів, які отримали нове лікування, зі тими, хто отримав старе. Ви навчите оцінювач Каплана — Мейєра для кожної групи пацієнтів і візуалізуєте їхні криві виживання поруч.

Датафрейм має назву recur і містить стовпці:

• age: вік пацієнта;
• treat: лікування, яке отримав пацієнт (0 — нове, 1 — старе);
• duration: тривалість після лікування в годинах;
• censor: чи цензурована подія;

Пакет pandas завантажено як pd, клас KaplanMeierFitter імпортовано з lifelines, а модуль pyplot імпортовано з matplotlib як plt.

Крім того, для вас уже створено екземпляр KaplanMeierFitter під назвою kmf і фігуру підграфіка ax.