From c56690bb70713da0eb9ae588da13c80dc7c9f54d Mon Sep 17 00:00:00 2001 From: Andrejh Date: Sun, 7 Feb 2021 19:04:36 +0100 Subject: [PATCH] add automatic labeling --- cara/montecarlo.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 19ad94f0..5a78c20c 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -440,9 +440,9 @@ def present_model(model: MCConcentrationModel, bins: int = 200) -> None: axs[0, 1].plot(ds, unmasked, 'k', label="Without mask") axs[0, 1].legend(loc="upper right") - # add the label automatically to the title of the plot ?? - # categories_particles = ("Breathing", "Speaking", "Shouting") ?? - axs[0, 1].set_title(r'Particle emissions') + categories_particles = ("Breathing", "Speaking", "Shouting") + axs[0, 1].set_title(r'Particle emissions - ' + f'{categories_particles[model.infected.expiratory_activity - 1]}') axs[0, 1].set_ylabel('Particle emission concentration [$cm^{-3}$]') axs[0, 1].set_xlabel(r'Diameter [$\mu$m]')