1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Phân tích sống sót bằng Python
Connected

Bài tập

Nhiệm kỳ của các thượng nghị sĩ

Bạn muốn so sánh liệu các thượng nghị sĩ từ Quebec có xu hướng phục vụ với độ dài nhiệm kỳ khác so với những người từ British Columbia hay không. Một cách đơn giản là vẽ các hàm sống sót Kaplan–Meier của họ trên cùng một đồ thị.

DataFrame chứa thông tin về thời lượng nhiệm kỳ của thượng nghị sĩ British Columbia có tên là bc_df. Nó gồm các cột:

  • name: tên của thượng nghị sĩ
  • diff_days: thời lượng thượng nghị sĩ tại nhiệm (tính theo ngày)
  • observed: liệu đã quan sát được thời điểm kết thúc nhiệm kỳ hay chưa (1 hoặc 0)

Đường cong sống sót của các thượng nghị sĩ Quebec đã được vẽ trên một đối tượng pyplot có tên senator_fig, và một thể hiện KaplanMeierFitter có tên kmf đã được tạo.

Ngoài ra, gói pandas đã được nạp với bí danh pd, lớp KaplanMeierFitter được import từ lifelines, và mô-đun pyplot đã được import từ matplotlib với bí danh plt.

Hướng dẫn

100 XP
  • Fit thể hiện KaplanMeierFitter là kmf với dữ liệu thời lượng nhiệm kỳ của các thượng nghị sĩ British Columbia.
  • Vẽ đường cong sống sót của các thượng nghị sĩ British Columbia trên cùng hình với đường cong của Quebec.
  • Hiển thị hình vẽ.