Comparar tratamientos para el dolor de los pacientes

Hay un experimento para un nuevo tratamiento contra el dolor, y quieres comparar a los pacientes que recibieron el tratamiento nuevo frente al antiguo. Ajustarás un estimador de Kaplan-Meier para cada conjunto de datos de pacientes y visualizarás sus curvas de supervivencia en paralelo.

El DataFrame se llama recur y contiene las columnas:

• age: la edad del paciente;
• treat: el tratamiento que recibió el paciente (0 si es el nuevo, 1 si es el antiguo);
• duration: la duración tras el tratamiento en horas;
• censor: si el evento está censurado;

El paquete pandas está cargado como pd, la clase KaplanMeierFitter está importada desde lifelines, y el módulo pyplot se ha importado desde matplotlib como plt.

Además, ya se han creado para ti una instancia de KaplanMeierFitter llamada kmf y una figura de subgráfica llamada ax.