НачатьНачать бесплатно

Построение кривой выживаемости

Вы — социолог, изучающий поведение осуждённых после освобождения. DataFrame prison содержит данные о 432 заключённых, освобождённых из тюрем штата Мэриленд в 1970-х годах; за каждым из них наблюдали в течение одного года после выхода на свободу. Интерес представляют два основных столбца:

  • week: количество недель, прошедших с момента освобождения
  • arrest: был ли осуждённый повторно арестован в течение года после освобождения

Функция выживаемости Каплана–Мейера уже подобрана для вас, а объект модели называется prison_kmf. Попробуйте визуализировать кривую выживаемости prison_kmf.

Библиотека matplotlib.pyplot импортирована как plt, модуль lifelines также уже импортирован.

Это упражнение является частью курса

Анализ выживаемости в Python

Посмотреть курс

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Plot the survival function
____

# Show the plot
____
Редактировать и запускать код