Vive la démocratie!

Plus tôt dans le chapitre, nous avons examiné et confirmé le problème de censure des données dans l'ensemble sur la durée des régimes. Il est maintenant temps de tracer votre première courbe de survie pour estimer la durée moyenne.

Votre tâche consiste à estimer une fonction de survie et à tracer une courbe de survie pour montrer les probabilités de survie. Pour estimer la fonction de survie, vous allez créer une instance de l'objet KaplanMeierFitter et l'ajuster aux colonnes de données appropriées. Les noms des paramètres vous sont fournis, et vous en apprendrez davantage sur l'ajustement des fonctions de survie dans les prochains chapitres.

Le paquet pandas est chargé sous le nom pd, le module matplotlib.pyplot est chargé sous le nom plt, et la classe KaplanMeierFitter est importée depuis lifelines. Le DataFrame des durées est chargé et stocké dans regime_durations.