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

improve labels of plots

Browse source
This commit is contained in:
Andrejh 2021-03-15 23:00:32 +01:00
parent 92ce654442
commit df40d70978
1 changed files with 10 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
cara/montecarlo.py
View file

@ -11,7 +11,7 @@ import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib.lines as mlines
from sklearn.neighbors import KernelDensity
TIME_STEP = 0.01
TIME_STEP = 0.001
USE_SCOEH = False
@ -688,7 +688,7 @@ def composite_plot_pi_vs_viral_load(baselines: typing.List[MCExposureModel], lab
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}$)', fontsize=12)
axs[0, 0].set_ylabel('Probability of infection\n$P(I|qID=60)$', fontsize=12)
axs[0, 0].set_ylabel('Probability of infection\n$P(\,\mathtt{I}\,|\,\mathrm{vl}\,)$', fontsize=12)
plt.suptitle(title, fontsize=12)
axs[0, 0].text(11, -0.01, '$(i)$')
@ -827,8 +827,8 @@ def generate_cdf_curves_vs_qr(masked: bool = False, samples: int = 200000, qid:
"""
fig, axs = plt.subplots(3, 1, sharex='all')
plt.suptitle("$F(qR|qID=$" + str(qid) + "$)$",fontsize=14, y=0.93)
fig.text(0.02, 0.5, 'Cumulative Distribution Function', va='center', rotation='vertical',fontsize=14)
plt.suptitle("$F(\mathrm{qR}|\mathrm{qID}=$" + str(qid) + "$)$",fontsize=16, y=0.93)
fig.text(0.02, 0.5, '', va='center', rotation='vertical',fontsize=16)
scenarios = [MCInfectedPopulation(
number=1,
@ -853,17 +853,17 @@ def generate_cdf_curves_vs_qr(masked: bool = False, samples: int = 200000, qid:
for i in range(3):
axs[i].hist(qr_values[3 * i:3 * (i + 1)], bins=2000, histtype='step',
color=colors, cumulative=True, range=(left, right))
axs[i].set_xlim(left, right)
color=colors, cumulative=True, range=(-7, 6))
axs[i].set_xlim(-6, 6)
axs[i].set_yticks([0, samples / 2, samples])
axs[i].set_yticklabels(['0.0', '0.5', '1.0'])
axs[i].yaxis.set_label_position("right")
axs[i].set_ylabel(activities[i],fontsize=12)
axs[i].set_ylabel(activities[i], fontsize=14)
axs[i].grid(linestyle='--')
axs[0].legend(handles=lines, loc='upper left')
plt.xlabel('qR (q h$^{-1}$)', fontsize=12)
tick_positions = np.arange(int(np.ceil(left)), int(np.ceil(right)), 2)
plt.xlabel('$\mathrm{qR}$ (q h$^{-1}$)', fontsize=16)
tick_positions = np.arange(-6, 6, 2)
plt.xticks(ticks=tick_positions, labels=['$\;10^{' + str(i) + '}$' for i in tick_positions])
fig.set_figheight(8)
@ -1258,6 +1258,6 @@ def plot_pi_vs_exposure_time(exp_models: typing.List[MCExposureModel], labels: t
plt.title('')
plt.xlabel(f'Travel time ({"min" if time_in_minutes else "h"})', fontsize=12)
plt.ylabel('Probability of infection\n$P(I|qID=60)$', fontsize=12)
plt.ylabel('Probability of infection\n$P(\,\mathtt{I}\,|\,\mathrm{vl}\,)$', fontsize=12)
plt.legend()
plt.show()