Kaplan-Meier estimator in Python – badass data science