From 16e705f5bb0d84ddafcff67c9778a6199f7e5822 Mon Sep 17 00:00:00 2001 From: Andrejh Date: Mon, 6 Sep 2021 17:00:30 +0200 Subject: [PATCH] increase sample size + add means of log(er) --- cara/results_paper.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/cara/results_paper.py b/cara/results_paper.py index 7b13d28f..4c5846fe 100644 --- a/cara/results_paper.py +++ b/cara/results_paper.py @@ -14,7 +14,7 @@ import matplotlib as mpl ######### Plot material ######### -SAMPLE_SIZE = 50000 +SAMPLE_SIZE = 500000 viral_loads = np.linspace(2, 12, 600) ############# Markers (for legend) ############# @@ -394,14 +394,16 @@ def build_breathing_legend(fig): label.set_position((-3*width, 0)) -def print_er_info(er: np.array): +def print_er_info(er: np.array, log_er: np.array): """ Prints statistical parameters of a given distribution of ER-values :param er: A numpy-array of the ER-values :return: Nothing, parameters are printed """ print(f"MEAN of ER = {np.mean(er)}\n" + f"MEAN of log ER = {np.mean(log_er)}\n" f"SD of ER = {np.std(er)}\n" + f"SD of log ER = {np.std(log_er)}\n" f"Median of ER = {np.quantile(er, 0.5)}\n") print(f"Percentiles of ER:")