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

Insert the PDF hist in the Pi vs vl plot

Browse source
This commit is contained in:
Andrejh 2021-02-21 16:08:03 +01:00
parent 6481b99948
commit 184981c400
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cara/montecarlo.py
View file

@ -569,6 +569,15 @@ def plot_pi_vs_viral_load(baselines: typing.Union[MCExposureModel, typing.List[M
plt.text(10.5, 0.80, '$vl_{crit2}$', fontsize=12,color='black') plt.text(10.5, 0.80, '$vl_{crit2}$', fontsize=12,color='black')
if labels is not None: if labels is not None:
plt.legend(labels) plt.legend(labels)
# this is an inset plot inside the main plot
a = plt.axes([.2, .25, .25, .2], facecolor='k')
#TODO - Markus can you plot the hist using the chosen model instead of hardcoding
# in the 'exposure_model' str?
plt.hist(shared_office_model[1].infection_probability()/100, bins=200)
plt.title('PDF',fontsize=10)
plt.xticks([0,0.5,1])
plt.yticks([])
plt.show() plt.show()