INNER CODE UNIT · Python
calc_stats
tuangauss/DataScienceProjects · Python/lincoln_estimate.py:27
def calc_stats(arr):
return (
np.mean(arr),
np.std(arr, ddof=1),
np.mean(arr) - 1.96*np.std(arr, ddof=1),
np.mean(arr) + 1.96*np.std(arr, ddof=1)
)
sims = [[0.3, 0.5], [0.6, 0.4], [0.7, 0.8], [0.9, 0.9]]
# create 2 lists, 1 of data frame of values, 1 of titles
res_arr = []
title_arr = []
for p in sims:
naive_estimates, lincoln_estimates = simulate(100, p[0], p[1], reps=100000)
naive_stats = calc_stats(naive_estimates)
lincoln_stats = calc_stats(lincoln_estimates)
naive_mean, naive_std = naive_stats[0], naive_stats[1]