def plot_conditional_probabilities(firsts, others): """ Part 3 - plot conditional probabilities for firsts and others For compatibility's sake, I'm limiting the range of weeks to the official solution's range of 35-46. """ weeks = xrange(35, 46) probabilities = {} for table in firsts, others: probabilities[table.pmf.name] = [] for week in weeks: new_pmf = utils.remaining_lifetime(table.pmf, week) probabilities[table.pmf.name].append(new_pmf.Prob(week)) for k, v in probabilities.iteritems(): pyplot.plot(weeks, v, label=k) myplot.Save(root='2_7', xlabel='weeks', ylabel=r'Prob{x $=$ weeks | x $\geq$ weeks}', title='Conditional Probability')
def _print_survival_analysis(week, table): new_pmf = utils.remaining_lifetime(table.pmf, week) print 'Probability that a live birth will be in Week {week_num} if not born '\ 'prior to Week {week_num} ({label}):'.format(week_num=week, label=table.pmf.name),\ new_pmf.Prob(week)