Побудуйте криву виживання

Ви — соціолог, який досліджує поведінку засуджених після звільнення. Датафрейм prison містить інформацію про 432 засуджених, звільнених із тюрем штату Меріленд у 1970-х роках, за якими спостерігали протягом року після звільнення. Є дві основні колонки:

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

Для вас уже підібрано оцінку функції виживання Каплана — Меєра, а об'єкт моделі має назву prison_kmf. Чи зможете ви візуалізувати криву виживання моделі prison_kmf?

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