Problem with stacking of markers in a legend

Community
1

I am plotting football ball markers as two separate scatter plots, one with a hexagon path and one with pentagon path. However, I can only get them to stack in the legend using a HandlerTuple in the legend. This requires knowledge of how legend handles and labels work and also needs to be done at the end of a plot.

If possible, I would like to hide the creation of the legend marker in a class method using a custom Handler and update the default handler map of the first scatter plot. However I have not been able to align the custom handler with the default legend markers created by scatter.

Here’s my problem:

Problem
Problem491×800 27.6 KB

I am not sure what to do next. Any help appreciated! Here’s the code for the various options I have tried so far: https://github.com/andrewRowlinson/mplsoccer/blob/0.0.4/tests/football_marker_legend.ipynb

2

I managed to fix it. _default_update_prop was overwriting the multiple colors.

class HandlerFootball(HandlerPathCollection):

def create_collection(self, orig_handle, sizes, offsets, transOffset):
    edgecolor = orig_handle.get_edgecolor()[0]
    facecolor = orig_handle.get_facecolor()[0]
    sizes = [size*0.249 for size in sizes]
    p = type(orig_handle)([football_hexagon_marker, football_pentagon_marker],
                          sizes=sizes,
                          offsets=offsets,
                          transOffset=transOffset,
                          facecolors=[facecolor, edgecolor],
                          edgecolors=edgecolor)
    return p

def _default_update_prop(self, legend_handle, orig_handle):
    facecolor = legend_handle.get_facecolor()
    edgecolor = legend_handle.get_edgecolor()
    legend_handle.update_from(orig_handle)
    legend_handle.set_facecolor(facecolor)
    legend_handle.set_edgecolor(edgecolor)
1 Like