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

add dynamic title in compare_concentration_curves

Browse source
This commit is contained in:
Andrejh 2021-02-26 21:29:12 +01:00
parent 4ad7a57477
commit e9621b7815
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cara/montecarlo.py
View file

@ -11,7 +11,7 @@ import matplotlib.pyplot as plt
import matplotlib.patches as patches import matplotlib.patches as patches
import matplotlib.lines as mlines import matplotlib.lines as mlines
from sklearn.neighbors import KernelDensity from sklearn.neighbors import KernelDensity
TIME_STEP = 0.05 TIME_STEP = 0.01
USE_SCOEH = False USE_SCOEH = False
@ -1043,6 +1043,7 @@ def plot_concentration_curve(model: MCExposureModel):
#print(np.max(lower)) #print(np.max(lower))
def compare_concentration_curves(exp_models: typing.List[MCExposureModel], labels: typing.List[str], def compare_concentration_curves(exp_models: typing.List[MCExposureModel], labels: typing.List[str],
title: str = 'Concentration profile',
colors: typing.Optional[typing.List[str]] = None, show_qd: bool = True) -> None: colors: typing.Optional[typing.List[str]] = None, show_qd: bool = True) -> None:
assert len(exp_models) == len(labels), "Different numbers of exposure models and labels" assert len(exp_models) == len(labels), "Different numbers of exposure models and labels"
assert all(model.concentration_model.virus.qID == exp_models[0].concentration_model.virus.qID assert all(model.concentration_model.virus.qID == exp_models[0].concentration_model.virus.qID
@ -1073,7 +1074,7 @@ def compare_concentration_curves(exp_models: typing.List[MCExposureModel], label
qds = [[np.trapz(c[:i + 1], times[:i + 1]) * factor for i in range(len(times))] qds = [[np.trapz(c[:i + 1], times[:i + 1]) * factor for i in range(len(times))]
for c, factor in zip(modified_concentrations, factors)] for c, factor in zip(modified_concentrations, factors)]
plt.suptitle("x scenario") plt.suptitle(title)
plt.xlabel("Time ($h$)", fontsize=12) plt.xlabel("Time ($h$)", fontsize=12)
plt.ylabel("Quantum concentration ($q\;m^{-3}$)\nmean values of $C(t)$", fontsize=12) plt.ylabel("Quantum concentration ($q\;m^{-3}$)\nmean values of $C(t)$", fontsize=12)
if show_qd: if show_qd: