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

Exercise

Test the PH assumption with KM curves

After learning about the proportional hazards assumption, you realize that the Cox PH model you built using prison DataFrame has not been tested. You will test the proportional hazards assumption for the categorical variable fin using Kaplan-Meier curves.

cph is the fitted Cox PH model using all columns, and it has been loaded. A KaplanMeierFitter class kmf has also been instantiated.

The CoxPHFitter class, the KaplanMeierFitter class, and the matplotlib.pyplot module have been imported. The pandas and numpy libraries are imported as pd and np, respectively.

Instructions 1/3

undefined XP
    1
    2
    3
  • Fit kmf to the data of convicts that did not receive financial assistance (fin: 0).
  • Plot the survival curve of kmf and assign the plot to a variable named ax.