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

change legend of present_model

Browse source
This commit is contained in:
markus 2021-02-17 17:34:25 +01:00
parent 60362addd1
commit b0646619ad
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cara/montecarlo.py
View file

@ -489,11 +489,12 @@ def present_model(model: MCConcentrationModel, bins: int = 200,
arrowprops={'arrowstyle': '<|-|>', 'ls': 'dashed'})
axs[1, 1].text(mean + std + 0.1, top * 0.92, 'Superspreader', fontsize=8)
# TODO: Markus, possible to replace 'patches' with just lines in the legend?
mean_patch = patches.Patch(color='grey', label='Mean')
median_patch = patches.Patch(color='black', label='Median')
std_patch = patches.Patch(color='lightgrey', linestyle='dashed', label='Standard deviations')
fig.legend(handles=(mean_patch, std_patch, median_patch), loc="upper left")
lines = [mlines.Line2D([], [], color=color, markersize=15, label=label, linestyle=style)
for color, label, style in zip(['grey', 'black', 'lightgrey'],
['Mean', 'Median', 'Standard deviation'],
['solid', 'solid', 'dashed'])]
fig.legend(handles=lines, loc="upper left")
print_qr_info(np.asarray(qRs))