Comparer les traitements de douleur chez les patients

On teste un nouveau traitement contre la douleur, et vous souhaitez comparer les patients ayant reçu le nouveau traitement à ceux ayant reçu l'ancien. Vous allez ajuster un estimateur de Kaplan-Meier pour chaque ensemble de données de patients et visualiser leurs courbes de survie côte à côte.

Le DataFrame s'appelle recur et contient les colonnes suivantes :

• age : l'âge du ou de la patient·e;
• treat : le traitement reçu par le ou la patient·e (0 pour le nouveau, 1 pour l'ancien);
• duration : la durée après le traitement, en heures;
• censor : indique si l'événement est censuré;

Le paquet pandas est chargé sous pd, la classe KaplanMeierFitter est importée de lifelines, et le module pyplot a été importé de matplotlib sous plt.

De plus, une instance de KaplanMeierFitter appelée kmf et une figure de sous-graphe appelée ax ont été créées pour vous.