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

add normalize_y_axis

Browse source
This commit is contained in:
markus 2021-03-03 16:15:08 +01:00
parent ffb0a14b75
commit d9fd316886
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cara/mc-output.py
View file

@ -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)

5
cara/montecarlo.py
View file

@ -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')