Хай живе демократія!

Раніше в цьому розділі ми дослідили й підтвердили проблему цензурування даних у наборі про тривалість режимів. Тепер час побудувати вашу першу криву виживання, щоб оцінити середню тривалість.

Ваше завдання — оцінити функцію виживання та побудувати криву виживання, щоб показати ймовірності виживання. Щоб оцінити функцію виживання, створіть екземпляр об'єкта KaplanMeierFitter і навчіть його на відповідних стовпцях даних. Назви параметрів уже надані — ви дізнаєтеся більше про підгонку функцій виживання в наступних розділах.

Пакет pandas завантажено як pd, модуль matplotlib.pyplot — як plt, а клас KaplanMeierFitter імпортовано з lifelines. Датафрейм з тривалостями завантажено й збережено як regime_durations.