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

add vertical lines showing the 'critical'vl values - Markus to make it more automatic

Browse source
This commit is contained in:
Andrejh 2021-02-16 22:20:58 +01:00
parent 7fd295930b
commit b6e2b7ce99
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cara/montecarlo.py
View file

@ -543,6 +543,10 @@ def plot_pi_vs_viral_load(baselines: typing.Union[MCExposureModel, typing.List[M
plt.ylabel('Percentage probability of infection') 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.xticks(ticks=[i for i in range(3, 13)], labels=['$10^{' + str(i) + '}$' for i in range(3, 13)])
plt.xlabel('Viral load') plt.xlabel('Viral load')
# add vertical lines for the critical viral loads for which pi= 0 or 100
# TODO Insert viral_load(Pi = 5) and viral_load(Pi = 95) instead of hard coded values
plt.vlines(x=(7, 11.5), ymin=0, ymax=100,
colors=('grey', 'black'), linestyles='dashed')
if labels is not None: if labels is not None:
plt.legend(labels) plt.legend(labels)