Dibuja una curva de supervivencia

Eres sociólogo y estudias cómo se comportan las personas condenadas tras su puesta en libertad. El DataFrame prison contiene información de 432 personas excarceladas de prisiones del estado de Maryland en los años 70 y a las que se hizo seguimiento durante un año tras su liberación. Hay dos columnas principales de interés:

• week: número de semanas transcurridas desde la liberación
• arrest: indica si la persona ha sido detenida de nuevo dentro del año posterior a la liberación

Ya se ha ajustado por ti una función de supervivencia de Kaplan-Meier y el objeto del modelo se llama prison_kmf. ¿Puedes visualizar la curva de supervivencia de prison_kmf?

La biblioteca matplotlib.pyplot está importada como plt y también se ha importado el módulo lifelines por ti.