Exercise

Plotting and comparing survival curves

Back at the heart clinic, you want to present some visualizations that showcase the differences in patient survival times to the Head of Research. You will plot the survival curves of patients with and without pericardial effusion side-by-side.

A KaplanMeierFitter object kmf is instantiated. DataFrames for patients with and without pericardial effusion are loaded and stored as has_pericardial_effusion and none_pericardial_effusion, respectively.

The KaplanMeierFitter class is imported for you. Additionally, the pandas package is loaded as pd, and the matplotlib.pyplot module is loaded as plt.

Instructions 1/2

undefined XP
    1
    2
  • Fit the estimator kmf to patients with pericardial effusion.
  • Plot the survival function from kmf and assign the plot object to a new variable surv_plot.