绘制生存曲线

您是一名社会学家，研究服刑人员获释后的行为。DataFrame prison 包含了 1970 年代从马里兰州监狱释放并在释放后 1 年内被跟踪随访的 432 名服刑人员的信息。主要包含两列：

• week：自释放以来经过的周数
• arrest：获释后 1 年内是否再次被逮捕

已经为您拟合了一个 Kaplan-Meier 生存函数，模型对象名为 prison_kmf。您能将 prison_kmf 的生存曲线可视化吗？

已为您导入 matplotlib.pyplot 库并命名为 plt，同时也已导入 lifelines 模块。