1. Learn
    2. /
  2. Cursuri
    3. /
  3. Analiza de supraviețuire în Python
Connected

exercițiu

Compararea tratamentelor pentru dureri musculare la pacienți

Există un experiment pentru un nou tratament al durerilor musculare și vrei să compari pacienții care au primit noul tratament față de cei care l-au primit pe cel vechi. Vei aplica un estimator Kaplan-Meier pe fiecare set de date de pacienți și vei vizualiza curbele de supraviețuire alăturat.

DataFrame-ul se numește recur și conține coloanele:

  • age: vârsta pacientului;
  • treat: tratamentul primit de pacient (0 pentru cel nou, 1 pentru cel vechi);
  • duration: durata după tratament, în ore;
  • censor: dacă evenimentul este cenzurat;

Pachetul pandas este importat ca pd, clasa KaplanMeierFitter este importată din lifelines, iar modulul pyplot din matplotlib este importat ca plt.

În plus, o instanță KaplanMeierFitter numită kmf și o figură cu subgrafic numită ax au fost deja create pentru tine.

Instrucțiuni 1/4

undefined XP
    1
    2
    3
    4
  • Creează o mască booleană pentru pacienții care au primit noul tratament.