預測再次被逮補的機率

你已經使用每位出獄受刑人的協變數（例如婚姻狀態、過去逮補紀錄等）建立了生存迴歸模型。現在有 10 名新的受刑人即將出獄，你要用這個模型來預測他們到再次被逮補的「中位數時間」。

新的 DataFrame 名為 prison_new，欄位與舊的 prison 相同。

WeibullAFTFitter 類別已匯入，先前訓練好的模型載入為 aft。pandas 與 numpy 分別以 pd 與 np 匯入。