Vẽ đường cong sống sót

Bạn là một nhà xã hội học nghiên cứu cách các phạm nhân cư xử sau khi được thả. DataFrame prison chứa thông tin từ 432 phạm nhân được thả khỏi các nhà tù bang Maryland vào những năm 1970 và được theo dõi trong một năm sau khi thả. Có hai cột chính cần quan tâm:

• week: số tuần kể từ khi được thả
• arrest: phạm nhân có bị bắt lại trong vòng một năm sau khi thả hay không

Một hàm sống sót Kaplan–Meier đã được khớp sẵn và đối tượng mô hình có tên prison_kmf. Bạn có thể trực quan hóa đường cong sống sót của prison_kmf không?

Thư viện matplotlib.pyplot đã được import với bí danh plt và mô-đun lifelines cũng đã được import sẵn cho bạn.