Сравнение методов лечения болезненности у пациентов

В рамках эксперимента изучается новый метод лечения болезненности. Вам нужно сравнить пациентов, получивших новое лечение, с теми, кто получил старое. Вы подберёте оценку Каплана–Мейера для каждой группы пациентов и визуализируете их кривые выживаемости рядом.

DataFrame называется recur и содержит следующие столбцы:

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

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

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