Как предыдущие аресты влияют на повторный арест?

В ходе изучения данных prison вы получаете новую информацию о каждом осуждённом, в частности количество судимостей до текущего заключения prio.

Вы хотите визуализировать, как различные значения prio влияют на базовую функцию выживаемости, смоделированную с помощью WeibullAFTFitter.

Класс WeibullAFTFitter импортирован, а новый экземпляр aft уже создан. Библиотека matplotlib.pyplot импортирована как plt, библиотеки pandas и numpy — как pd и np соответственно. При необходимости используйте консоль для изучения датафрейма и названий его столбцов.