วาดกราฟ Survival Curve

คุณเป็นนักสังคมวิทยาที่ศึกษาพฤติกรรมของผู้ต้องขังหลังได้รับการปล่อยตัว DataFrame prison มีข้อมูลของผู้ต้องขัง 432 คนที่ได้รับการปล่อยตัวจากเรือนจำรัฐแมริแลนด์ในช่วงทศวรรษ 1970 และได้รับการติดตามผลเป็นเวลา 1 ปีหลังปล่อยตัว คอลัมน์หลักที่ต้องสนใจมี 2 คอลัมน์ ได้แก่

• week: จำนวนสัปดาห์ที่ผ่านไปนับตั้งแต่ได้รับการปล่อยตัว
• arrest: ระบุว่าผู้ต้องขังถูกจับกุมซ้ำภายใน 1 ปีหลังปล่อยตัวหรือไม่

ได้มีการ Fit Kaplan-Meier Survival Function ให้แล้ว และ Object ของโมเดลนี้ใช้ชื่อว่า prison_kmf ลองสร้างกราฟ Survival Curve ของ prison_kmf กันเลย

ไลบรารี matplotlib.pyplot ถูก Import ไว้เป็น plt และ Module lifelines ก็ถูก Import ไว้ให้แล้วเช่นกัน