Сроки пребывания сенаторов в должности

Вы хотите сравнить, отличаются ли сроки пребывания в должности у сенаторов из Квебека и Британской Колумбии. Один из простых способов — отобразить их функции выживаемости Каплана–Мейера на одном графике.

DataFrame с информацией о сроках пребывания сенаторов из Британской Колумбии называется bc_df. Он содержит следующие столбцы:

name: имя сенатора
diff_days: продолжительность пребывания сенатора в должности
observed: наблюдался ли конец срока (1 или 0)

Кривая выживаемости сенаторов из Квебека уже построена на объекте pyplot с именем senator_fig, а экземпляр KaplanMeierFitter с именем kmf уже создан.

Кроме того, пакет pandas импортирован как pd, класс KaplanMeierFitter импортирован из lifelines, а модуль pyplot из matplotlib импортирован как plt.

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

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

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

Инструкции к упражнению

Обучите экземпляр KaplanMeierFitter kmf на данных о сроках пребывания сенаторов из Британской Колумбии.
Постройте кривую выживаемости сенаторов из Британской Колумбии на том же графике, что и кривую для сенаторов из Квебека.
Отобразите график.

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

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

# Fit Kaplan-Meier estimator
____(____, ____, label='British Columbia')

# Plot survival function on senator_fig
____

# Display the figure
____

Редактировать и запускать код