from __future__ import (absolute_import, division,
                        print_function, unicode_literals)
import six
from matplotlib import pyplot as plt
import numpy as np
from matplotlib.legend_handler import HandlerLineCollection
import matplotlib.collections as mcol
from matplotlib.lines import Line2D


class HandlerDashedLines(HandlerLineCollection):
    """
    Custom Handler for LineCollection instances.
    """
    def create_artists(self, legend, orig_handle,
                       xdescent, ydescent, width, height, fontsize, trans):
        # figure out how many lines there are
        numlines = len(orig_handle.get_segments())
        xdata, xdata_marker = self.get_xdata(legend, xdescent, ydescent,
                                             width, height, fontsize)
        leglines = []
        # divide the vertical space where the lines will go
        # into equal parts based on the number of lines
        ydata = ((height) / (numlines + 1)) * np.ones(xdata.shape, float)
        # for each line, create the line at the proper location
        # and set the dash pattern
        for i in range(numlines):
            legline = Line2D(xdata, ydata * (numlines - i) - ydescent)
            self.update_prop(legline, orig_handle, legend)
            # set color, dash pattern, and linewidth to that
            # of the lines in linecollection
            try:
                color = orig_handle.get_colors()[i]
            except IndexError:
                color = orig_handle.get_colors()[0]
            try:
                dashes = orig_handle.get_dashes()[i]
            except IndexError:
                dashes = orig_handle.get_dashes()[0]
            try:
                lw = orig_handle.get_linewidths()[i]
            except IndexError:
                lw = orig_handle.get_linewidths()[0]
            if dashes[0] is not None:
                legline.set_dashes(dashes[1])
            legline.set_color(color)
            legline.set_transform(trans)
            legline.set_linewidth(lw)
            leglines.append(legline)
        return leglines

x = np.linspace(0, 5, 100)

plt.figure()
colors = plt.rcParams['axes.prop_cycle'].by_key()['color'][:5]
styles = ['solid', 'dashed', 'dashed', 'dashed', 'solid']
lines = []
for i, color, style in zip(range(5), colors, styles):
    plt.plot(x, np.sin(x) - .1 * i, c=color, ls=style)


# make proxy artists
# make list of one line -- doesn't matter what the coordinates are
line = [[(0, 0)]]
# set up the proxy artist
lc = mcol.LineCollection(5 * line, linestyles=styles, colors=colors)
# create the legend
plt.legend([lc], ['multi-line'], handler_map={type(lc): HandlerDashedLines()},
           handlelength=2.5, handleheight=3)

plt.show() 

　

Traceback (most recent call last):
  File "M:\______\legend_demo5.py", line 68, in 
    plt.legend([lc], ['multi-line'], handler_map={type(lc): HandlerDashedLines()},
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\pyplot.py", line 2710, in legend
    return gca().legend(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\axes\_axes.py", line 318, in legend
    self.legend_ = mlegend.Legend(self, handles, labels, **kwargs)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\_api\deprecation.py", line 454, in wrapper
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\legend.py", line 583, in __init__
    self._init_legend_box(handles, labels, markerfirst)
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\legend.py", line 867, in _init_legend_box
    handle_list.append(handler.legend_artist(self, orig_handle,
                       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Users\kamif\AppData\Local\Programs\Python\Python311\Lib
    \site-packages\matplotlib\legend_handler.py", line 130, in legend_artist
    artists = self.create_artists(legend, orig_handle,
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "M:\______\legend_demo5.py", line 24, in create_artists
    ydata = ((height) / (numlines + 1)) * np.ones(xdata.shape, float)
                                                  ^^^^^^^^^^^
AttributeError: 'list' object has no attribute 'shape' 

Traceback (most recent call last):
  File "E:\______\legend_demo5.py", line 68, in 
    plt.legend([lc], ['multi-line'], handler_map={type(lc): HandlerDashedLines()},
  File "C:\Program Files\Python312\Lib\site-packages\matplotlib\pyplot.py", 
    line 3372, in legend
    return gca().legend(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Program Files\Python312\Lib\site-packages\matplotlib\axes\_axes.py", 
    line 323, in legend
    self.legend_ = mlegend.Legend(self, handles, labels, **kwargs)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Program Files\Python312\Lib\site-packages\matplotlib\legend.py", 
    line 563, in __init__
    self._init_legend_box(handles, labels, markerfirst)
  File "C:\Program Files\Python312\Lib\site-packages\matplotlib\legend.py", 
    line 913, in _init_legend_box
    handle_list.append(handler.legend_artist(self, orig_handle,
                       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "C:\Program Files\Python312\Lib\site-packages\matplotlib\legend_handler.py", 
    line 129, in legend_artist
    artists = self.create_artists(legend, orig_handle,
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "E:\______\legend_demo5.py", line 24, in create_artists
    ydata = ((height) / (numlines + 1)) * np.ones(xdata.shape, float)
                                                  ^^^^^^^^^^^
AttributeError: 'list' object has no attribute 'shape' 

• 2023/11/28 Ver=1.04 Python 3.12.0 (matplotlib 3.8.1)で確認
• 2023/11/28 Ver=1.04 Python 3.11.6 (matplotlib 3.7.1)で確認
• 2023/04/03 Ver=1.03 Python 3.11.2 で確認
• 2020/10/31 Ver=1.01 Python 3.7.8 で確認
• 2018/12/03 Ver=1.01 初版リリース

Top