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

Connected

Exercise

Model prison data with Weibull

You are a sociologist studying the time it takes for convicts to be rearrested after their release. The DataFrame prison contains information from 432 convicts who were released from Maryland state prisons in the 1970s and who were followed up with one year after release.

You will model the survival function of time-to-arrest using the Weibull model.

The matplotlib.pyplot library is imported as plt, and the pandas and numpy libraries are imported as pd and np, respectively. Use the console to explore the DataFrame and its column names as needed.

Instructions 1/4

undefined XP
    1
    2
    3
    4
  • Import the WeibullFitter class from lifelines.
  • Create an instance of WeibullFitter called wb.