From b6db943bc2be6296cabdb02bc015a18cecb48852 Mon Sep 17 00:00:00 2001 From: Andrejh Date: Tue, 2 Mar 2021 14:52:21 +0100 Subject: [PATCH 1/2] lables pi vs vl --- cara/montecarlo.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 03d6092d..f880d72b 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -558,9 +558,9 @@ def plot_pi_vs_viral_load(baselines: typing.Union[MCExposureModel, typing.List[M plt.fill_between(viral_loads, lower_percentiles, upper_percentiles, alpha=0.2) plt.title(title) - plt.ylabel('Probability of infection (%)\n$P(I|qID=60)$') + plt.ylabel('Probability of infection (%)', fontsize=14) plt.xticks(ticks=[i for i in range(3, 13)], labels=['$10^{' + str(i) + '}$' for i in range(3, 13)]) - plt.xlabel('Viral load (RNA copies mL$^{-1}$)\n$vl$') + plt.xlabel('Viral load (RNA copies mL$^{-1}$)', fontsize=14) # add vertical lines for the critical viral loads for which pi= 5 or 95 if len(baselines) == 1: @@ -579,8 +579,8 @@ def plot_pi_vs_viral_load(baselines: typing.Union[MCExposureModel, typing.List[M plt.vlines(x=(left, right), ymin=0, ymax=1, colors=('grey', 'grey'), linestyles='dotted') - plt.text(left - 1.1, 0.80, '$vl_{0.05}$', fontsize=12,color='black') - plt.text(right + 0.1, 0.80, '$vl_{0.95}$', fontsize=12,color='black') + plt.text(left - 1.1, 0.80, '$vl_{0.05}$', fontsize=14,color='black') + plt.text(right + 0.1, 0.80, '$vl_{0.95}$', fontsize=14,color='black') # add 3 shaded areas plt.axvspan(3, left, alpha=0.1, color='limegreen') plt.axvspan(left, right, alpha=0.1, color='orange') From be6d87035b621a1e8a8de913bca770aae0c1f53a Mon Sep 17 00:00:00 2001 From: Andrejh Date: Tue, 2 Mar 2021 15:39:55 +0100 Subject: [PATCH 2/2] minor adjustments in pi vs vl --- cara/montecarlo.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index f880d72b..8516a600 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -687,9 +687,9 @@ def composite_plot_pi_vs_viral_load(baselines: typing.List[MCExposureModel], lab axs[1, 0].set_xticks([i for i in range(2, 13, 2)]) axs[1, 0].set_xticklabels(['$10^{' + str(i) + '}$' for i in range(2, 13, 2)]) axs[1, 0].set_xlim(2, 12) - axs[1, 0].set_xlabel('Viral load (RNA copies mL$^{-1}$)\n$vl$') - axs[0, 0].set_ylabel('Probability of infection (%)\n$P(I|qID=60)$') - plt.suptitle(title) + axs[1, 0].set_xlabel('Viral load (RNA copies mL$^{-1}$)', fontsize=12) + axs[0, 0].set_ylabel('Probability of infection (%)\n$P(I|qID=60)$', fontsize=12) + plt.suptitle(title, fontsize=12) axs[0, 0].text(11, -0.01, '$(i)$') axs[1, 0].text(11, axs[1, 0].get_ylim()[1] * 0.8, '$(ii)$') @@ -704,12 +704,12 @@ def composite_plot_pi_vs_viral_load(baselines: typing.List[MCExposureModel], lab break for i, (crit, color) in enumerate(zip(crits, colors)): - axs[0, 0].text(2.5, 0.4 - i * 0.1, f'x $vl_{"{0.95}"}=' + '10^{' + str(np.round(crits[i], 1)) + '}$', fontsize=10, color=color) + axs[0, 0].text(2.5, 0.45 - i * 0.1, f'x $vl_{"{0.95}"}=' + '10^{' + str(np.round(crits[i], 1)) + '}$', fontsize=10, color=color) axs[0, 0].plot(crits[i], 0.95, 'x', color=color) if show_lines: axs[0, 0].hlines([0.5], colors=['lightgrey'], linestyles=['dashed'], xmin=2, xmax=12) - axs[0, 0].text(9.7, 0.52, "$P(I) = 0.5$", color='lightgrey') + axs[0, 0].text(9.7, 0.52, "$P(I) = 0.5$", color='grey') middle_positions = [] for line in lines: for i, point in enumerate(line):