From b0646619ad133ec90a5a2fae35fa21157bba8175 Mon Sep 17 00:00:00 2001 From: markus Date: Wed, 17 Feb 2021 17:34:25 +0100 Subject: [PATCH] change legend of present_model --- cara/montecarlo.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/cara/montecarlo.py b/cara/montecarlo.py index 7a15b68c..d45acc4a 100644 --- a/cara/montecarlo.py +++ b/cara/montecarlo.py @@ -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))