Merge branch 'feature/virions_plot' of https://gitlab.cern.ch/cara/cara into feature/virions_plot

This commit is contained in:
Luis Aleixo 2021-09-06 17:14:29 +02:00
commit 83caf48282

View file

@ -120,10 +120,10 @@ def exposure_model_from_vl_breathing_cn():
ax.plot(viral_loads, er_means, color=cmap.to_rgba(
cn, alpha=0.75), linewidth=1, ls='--')
plt.text(viral_loads[int(len(viral_loads)*0.9)], 10**4.2,
r"$\mathbf{c_{n,B}=0.06}$", color=cmap.to_rgba(cn), size='small')
r"$\mathbf{c_{n,B}=0.06}$", color=cmap.to_rgba(cn), fontesize=12)
fig.colorbar(cmap, ticks=[0.01, 0.1, 0.25, 0.5],
label="Particle emission concentration, ${c_{n,B}}$")
label="Particle emission concentration, ${c_{n,B}}$", fontesize=14)
ax.set_yscale('log')
############# Coleman #############
@ -185,7 +185,7 @@ def exposure_model_from_vl_talking():
build_talking_legend(fig)
############ Plot ############
plt.title('Exhaled virions while talking for 15min',
plt.title('Exhaled virions while speaking (vocalization) for 15min',
fontsize=16, fontweight="bold")
plt.ylabel(
'Aerosol viral load, $\mathrm{vl_{out}}$\n(RNA copies)', fontsize=14)
@ -230,10 +230,10 @@ def exposure_model_from_vl_talking_cn():
ax.plot(viral_loads, er_means, color=cmap.to_rgba(
cn, alpha=0.75), linewidth=1, ls='--')
plt.text(viral_loads[int(len(viral_loads)*0.93)], 10**5.5,
r"$\mathbf{c_{n,L}=0.2}$", color=cmap.to_rgba(cn), size='small')
r"$\mathbf{c_{n,L}=0.2}$", color=cmap.to_rgba(cn), fontesize=12)
fig.colorbar(cmap, ticks=[0.01, 0.5, 1.0, 2.0],
label="Particle emission concentration, ${c_{n,L}}$")
label="Particle emission concentration, ${c_{n,L}}$", fontesize=14)
ax.set_yscale('log')
############# Coleman #############
@ -243,7 +243,7 @@ def exposure_model_from_vl_talking_cn():
build_talking_legend(fig)
############ Plot ############
plt.title('Exhaled virions while talking for 15min',
plt.title('Exhaled virions while speaking (vocalization)\nfor 15min',
fontsize=16, fontweight="bold")
plt.ylabel(
'Aerosol viral load, $\mathrm{vl_{out}}$\n(RNA copies)', fontsize=14)
@ -414,14 +414,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:")