1. Learn
    2. /
  2. Courses
    3. /
  3. Survival Analysis in Python

Exercise

Patient soreness treatment

You are studying a treatment for muscle soreness recurrence in patients. You want to use the Kaplan-Meier estimator to visualize the rate at which patients receiving the treatment encounter soreness episodes.

The DataFrame is called recur and contains columns

  • age: the age of the patient;
  • time_0: the time that the patient received the treatment;
  • time_1: the last time of an event, or the last time you measured a censored patient;
  • censor: whether the event is censored;

The pandas package is loaded as pd, the KaplanMeierFitter class is imported from lifelines, and the pyplot module has been import from matplotlib as plt.

Instructions 1/4

undefined XP
    1
    2
    3
    4
  • Create a new column for the duration of the observation and call it duration.