Lunga vita alla democrazia!

All’inizio del capitolo abbiamo esplorato e confermato il problema della censura dei dati nel dataset sulle durate dei regimi. Ora è il momento di costruire la tua prima curva di sopravvivenza per stimare la durata media.

Il tuo compito è stimare una funzione di sopravvivenza e tracciare una curva di sopravvivenza per mostrare le probabilità di sopravvivenza. Per stimare la funzione di sopravvivenza, creerai un’istanza dell’oggetto KaplanMeierFitter e lo adatterai alle colonne di dati appropriate. I nomi dei parametri sono già forniti: imparerai di più su come adattare le funzioni di sopravvivenza nei prossimi capitoli.

Il pacchetto pandas è caricato come pd, il modulo matplotlib.pyplot è caricato come plt e la classe KaplanMeierFitter è importata da lifelines. Il DataFrame delle durate è caricato e salvato come regime_durations.