1. 학습
    2. /
  2. 강의
    3. /
  3. Python에서의 Survival Analysis
Connected

연습 문제

생존 곡선 그리기

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

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

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

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

지침 1/2

undefined XP
    1
    2
  • prison_kmf의 생존 곡선을 그리세요.
  • 생존 곡선 플롯을 화면에 표시하세요.