Eine Überlebenskurve zeichnen

Du bist Soziologe und untersuchst, wie sich Strafgefangene nach ihrer Entlassung verhalten. Der DataFrame prison enthält die Daten von 432 Strafgefangenen, die in den 1970er Jahren aus den staatlichen Gefängnissen in Maryland entlassen wurden und ein Jahr nach der Entlassung weiterverfolgt wurden. Es gibt zwei wichtige relevante Spalten:

• week: die Anzahl der seit der Entlassung verstrichenen Wochen
• arrest: ob der Strafgefangene innerhalb eines Jahres nach seiner Entlassung erneut verhaftet wurde

Eine Kaplan-Meier-Überlebensfunktion wurde für dich angepasst und das Modellobjekt heißt prison_kmf. Kannst du die Überlebenskurve von prison_kmf visualisieren?

Die Bibliothek matplotlib.pyplot wird als plt importiert und das Modul lifelines wurde ebenfalls für dich importiert.