생존 곡선 그리기

여러분은 출소 이후 전과자의 행동을 연구하는 사회학자입니다. DataFrame prison에는 1970년대 메릴랜드 주 교도소에서 출소한 432명의 전과자 정보를 담고 있으며, 출소 후 1년 동안의 추적 결과가 포함되어 있습니다. 주요 관심 열은 다음 두 가지입니다:

• week: 출소 후 경과한 주 수
• arrest: 출소 후 1년 이내에 재구속되었는지 여부

이미 Kaplan–Meier 생존함수를 적합해 두었고, 모델 객체 이름은 prison_kmf입니다. prison_kmf의 생존 곡선을 시각화해 볼까요?

matplotlib.pyplot은 plt로, lifelines 모듈도 이미 임포트되어 있습니다.