pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update concentration plot

Browse source
This commit is contained in:
markus 2021-02-04 13:52:19 +01:00
parent 57839595fc
commit a1c6a431af
1 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cara/montecarlo.py
View file

@ -410,9 +410,18 @@ def present_model(model: MCConcentrationModel, bins: int = 30) -> None:
axs[0, 0].set_title('Viral load in sputum')
axs[0, 0].set_xlabel('Viral load [log10(RNA copies / mL)]')
ds = np.linspace(0.1, 30, 2000)
concentrations = model.infected._concentration_distribution()(ds)
axs[0, 1].plot(ds, concentrations)
ds = np.linspace(0.1, 15, 2000)
unmasked = model.infected._concentration_distribution_without_mask()(ds)
masked = model.infected._concentration_distribution_with_mask()(ds)
if model.infected.masked:
axs[0, 1].plot(ds, masked, 'g', label="With mask")
axs[0, 1].plot(ds, unmasked, 'r--', label="Without mask")
axs[0, 1].legend(loc="upper right")
else:
axs[0, 1].plot(ds, masked, 'g--', label="With mask")
axs[0, 1].plot(ds, unmasked, 'r', label="Without mask")
axs[0, 1].legend(loc="upper right")
axs[0, 1].set_title(r'Particle concentration vs diameter')
axs[0, 1].set_ylabel('Concentration [cm^-3]')
axs[0, 1].set_xlabel(r'Diameter [$\mu$m]')