From d78a46bc0b800472cd45fe9552f59e75ba085546 Mon Sep 17 00:00:00 2001 From: markus Date: Thu, 11 Feb 2021 12:44:14 +0100 Subject: [PATCH] add labels to pi_vs_viral_load plot --- cara/montecarlo.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 03a985a3..77ddca55 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -531,6 +531,10 @@ def plot_pi_vs_viral_load(baseline: MCExposureModel, samples_per_vl: int = 20000 plt.plot(viral_loads, pi_medians) plt.fill_between(viral_loads, lower_percentiles, upper_percentiles, alpha=0.2) + plt.title('Probability of infection vs viral load') + plt.ylabel('Percentage probability of infection') + plt.xticks(ticks=[i for i in range(3, 13)], labels=['$10^{' + str(i) + '}$' for i in range(3, 13)]) + plt.xlabel('Viral load in sputum') plt.show()