def main():
pregnancies = first.get_pregnancies_dataset()
live_pregnancies = [preg for preg in pregnancies.records if preg.outcome == 1]
firsts, subsequents = first.split_one_vs_all(live_pregnancies,
birthord=1)
period_first = first.get_values_for_field(firsts, 'prglength')
period_subsequent = first.get_values_for_field(subsequents, 'prglength')
var_first = thinkstats.Var(period_first)
var_second = thinkstats.Var(period_subsequent)
print('Variance of the first children gestation periods: {} weeks^2'
.format(var_first))
print('\tAnd the sd is {} weeks.'.format(math.sqrt(var_first)))
print('Variance of the subsequent children gestation periods: {} weeks^2'
.format(var_second))
print('\tAnd the sd is {} weeks.'.format(math.sqrt(var_second)))
weights = get_pumpkin_weights()
print('Finding the variance and mean for these pumpkins: {}'
.format(weights))
mu, var = pumpkins(weights)
sd = math.sqrt(var)
print('\tmean = {}\tvariance = {}\tsd = {}'.format(mu, var, sd))
bar_chart(pregnancies)
plot_pregnancy_pmf(pregnancies)
plot_pregnancies_pmf_differences(pregnancies)
def main():
live = first.get_live_pregnancies(first.get_pregnancies_dataset())
firsts, others = first.split_one_vs_all(live, birthord=1)
pmf_firsts = Pmf.MakePmfFromList(item.prglength for item in firsts)
pmf_others = Pmf.MakePmfFromList(item.prglength for item in others)
pmf_all = Pmf.MakePmfFromList(item.prglength for item in live)
birth_types = ('first babies', 'subsequent babies', 'all live births')
prob_types = ('early', 'on time', 'late')
for i, pmf in enumerate((pmf_firsts, pmf_others, pmf_all)):
for j, f in enumerate((prob_early, prob_on_time, prob_late)):
print('The probability of {} being born {} is {} %'.format(
birth_types[i], prob_types[j], f(pmf) * 100
))
risks = get_relative_risks(pmf_firsts, pmf_others)
print()
for prob_type in prob_types:
print('The relative risk of first babies to others arriving {} is {}'
.format(prob_type, risks[prob_type]))
Get the probability that a birth will happen in week |week|, given that
it has not happened before it.
Args:
pmf(Pmf.Pmf): The PMF of pregnancies.
'''
my_pmf = pmf.Copy()
for value in my_pmf.Values():
if value < week:
my_pmf.Remove(value)
my_pmf.Normalize()
return my_pmf.Prob(week)
if __name__ == '__main__':
live_pregnancies = get_live_pregnancies(get_pregnancies_dataset())
firsts, others = split_one_vs_all(live_pregnancies,
birthord=1)
first_durations = [record.prglength for record in firsts]
others_durations = [record.prglength for record in others]
first_pmf = Pmf.MakePmfFromList(first_durations)
others_pmf = Pmf.MakePmfFromList(others_durations)
fig = plt.figure()
sp = fig.add_subplot(111)
# weeks1, weeks2 = (range(min(first_durations), max(first_durations)),
# range(min(others_durations), max(others_durations)))
weeks1, weeks2 = map(lambda t: range(*t), ((36, 45), (36, 45)))
probs_first = [probability_of_birth_in_week(first_pmf, week)
for week in weeks1]
probs_others = [probability_of_birth_in_week(others_pmf, week)
for week in weeks2]
