Nakresli křivku přežití

Jsi sociolog/socioložka, který/á zkoumá, jak se chovají odsouzení po propuštění z vězení. DataFrame prison obsahuje informace o 432 odsouzených, kteří byli propuštěni z věznic státu Maryland v 70. letech a byli sledováni jeden rok po propuštění. Jsou zde dva hlavní sloupce:

• week: počet týdnů, které uplynuly od propuštění
• arrest: zda byl odsouzený znovu zatčen do jednoho roku od propuštění

Kaplan-Meierova funkce přežití už byla za tebe natrénována a objekt modelu se jmenuje prison_kmf. Dokážeš vizualizovat křivku přežití modelu prison_kmf?

Knihovna matplotlib.pyplot je naimportována jako plt a modul lifelines je také k dispozici.