Behandelingen voor spierpijn bij patiënten vergelijken

Er loopt een experiment met een nieuwe behandeling tegen spierpijn, en je wilt patiënten vergelijken die de nieuwe behandeling kregen met degenen die de oude kregen. Je past een Kaplan-Meier-schatting toe op elke set patiëntgegevens en visualiseert hun overlevingscurves naast elkaar.

De DataFrame heet recur en bevat de kolommen:

• age: de leeftijd van de patiënt;
• treat: de behandeling die de patiënt kreeg (0 voor nieuw, 1 voor oud);
• duration: de tijd na behandeling in uren;
• censor: of de gebeurtenis gecensureerd is;

Het pakket pandas is geladen als pd, de klasse KaplanMeierFitter is geïmporteerd uit lifelines, en de module pyplot is uit matplotlib geïmporteerd als plt.

Daarnaast zijn er alvast een KaplanMeierFitter-instantie kmf en een subplot-figuur ax voor je aangemaakt.