diff --git a/cara/mc-output.py b/cara/mc-output.py index 13aa23d1..552b8a68 100644 --- a/cara/mc-output.py +++ b/cara/mc-output.py @@ -16,7 +16,8 @@ plot_pi_vs_exposure_time(ski_cabin_model_baseline_exposure_time, ['model1', 'mod colors=['red', 'green'], linestyles=['dotted', 'dashed'], points=20, - time_in_minutes=True) + time_in_minutes=True, + normalize_y_axis=True) # compare_viruses_qr(violins=True) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 1125a675..5f6f404f 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -1193,7 +1193,7 @@ def compare_viruses_qr(violins: bool = True) -> None: def plot_pi_vs_exposure_time(exp_models: typing.List[MCExposureModel], labels: typing.List[str], colors: typing.Optional[typing.List] = None, linestyles: typing.Optional[typing.List[str]] = None, - points: int = 50, time_in_minutes: bool = False) -> None: + points: int = 50, time_in_minutes: bool = False, normalize_y_axis: bool = False) -> None: conc_models = [m.concentration_model for m in exp_models] if colors is None: colors = [None for _ in exp_models] @@ -1235,6 +1235,9 @@ def plot_pi_vs_exposure_time(exp_models: typing.List[MCExposureModel], labels: t for i, pi in enumerate(pis): plt.plot(times, pi, color=colors[i], linestyle=linestyles[i], label=labels[i]) + if normalize_y_axis: + plt.ylim((0, 1)) + plt.title('TITLE HERE') plt.xlabel(f'XLABEL HERE ({"min" if time_in_minutes else "h"})') plt.ylabel('YLABEL HERE')