Vida longa à democracia!

No início do capítulo, você explorou e confirmou o problema de censura no conjunto de dados de duração de regimes. Agora é hora de construir sua primeira curva de sobrevivência para estimar a duração média.

Sua tarefa é estimar uma função de sobrevivência e traçar uma curva de sobrevivência para mostrar as probabilidades de sobrevivência. Para estimar a função de sobrevivência, você vai criar uma instância do objeto KaplanMeierFitter e ajustá-la às colunas de dados apropriadas. Os nomes dos parâmetros foram fornecidos, e você aprenderá mais sobre o ajuste de funções de sobrevivência nos próximos capítulos.

O pacote pandas está carregado como pd, o módulo matplotlib.pyplot está carregado como plt, e a classe KaplanMeierFitter foi importada de lifelines. O DataFrame de durações está carregado e armazenado como regime_durations.