Survival analysis in Python