Desenhar uma curva de sobrevivência

Você é um sociólogo estudando como condenados se comportam após a libertação. O DataFrame prison contém informações de 432 pessoas libertadas de prisões estaduais de Maryland na década de 1970 e acompanhadas por um ano após a soltura. Há duas colunas principais de interesse:

• week: número de semanas desde a libertação
• arrest: se a pessoa foi presa novamente dentro de um ano após a libertação

Uma função de sobrevivência de Kaplan-Meier já foi ajustada para você e o objeto do modelo se chama prison_kmf. Você consegue visualizar a curva de sobrevivência de prison_kmf?

A biblioteca matplotlib.pyplot foi importada como plt e o módulo lifelines também já foi importado para você.