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

add tertiary axis and labels

Browse source
This commit is contained in:
markus 2021-02-24 14:32:37 +01:00
parent fdf4f38bbf
commit fac33a5cf1
1 changed files with 20 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
cara/montecarlo.py
View file

@ -1037,18 +1037,32 @@ def compare_concentration_curves(exp_models: typing.List[MCExposureModel], label
times = np.arange(start, stop, TIME_STEP)
concentrations = [[np.median(model.concentration_model.concentration(t)) for t in times] for model in exp_models]
fig, ax = plt.subplots()
for c, label, color in zip(concentrations, labels, colors):
plt.plot(times, c, label=label, color=color)
ax.plot(times, c, label=label, color=color)
ax.legend(loc='upper left')
plt.legend()
factors = [0.6 * model.exposed.activity.inhalation_rate * (1 - model.exposed.mask.η_inhale) for model in exp_models]
qds = [[np.trapz(c[:i + 1], times[:i + 1]) * factor for i in range(len(times))]
for c, factor in zip(concentrations, factors)]
plt.suptitle("SUPTITLE HERE")
plt.xlabel("XLABEL HERE")
plt.ylabel("YLABEL HERE")
if show_qd:
plt.twinx()
for qd, label, color in zip(qds, labels, colors):
plt.plot(times, qd, label='qD - ' + label, color=color, linestyle='dotted')
plt.legend()
ax1 = ax.twinx()
for qd, label, color in zip(qds, labels, colors):
ax1.plot(times, qd, label='qD - ' + label, color=color, linestyle='dotted')
ax1.set_ylabel('qD LABEL HERE')
ax2 = ax.twinx()
ax2.spines["right"].set_position(("axes", 1.2))
ax2.set_ylabel('RNA LABEL HERE')
ax2.set_ylim(tuple(y * exp_models[0].concentration_model.virus.qID for y in ax1.get_ylim()))
ax1.legend(loc='upper right')
plt.tight_layout()
plt.show()