Блог

Мне нужно создать легенду для сегмента линии, который имеет другой маркер внизу и вверху строки. Я могу создать легенду с повторением 1 символа маркера, но не с двумя разными маркерами на каждом конце.

Вот воспроизводимый пример.

 from matplotlib import pyplot as plt
import numpy as np

#Create some data
x = np.arange(0,11)
y1 = np.sqrt(x/2.)
y2 = x


plt.figure(figsize=(8,8))
ax = plt.subplot(111)

#Plot the lines
for i,x_ in zip(range(11),x):
    ax.plot([x_,x_],[y1[i],y2[i]],c='k')
    
#Plot the end points    
ax.scatter(x,y1,marker='s',c='r',s=100,zorder=10)
ax.scatter(x,y2,marker='o',c='r',s=100,zorder=10)

ax.plot([],[],c='k',marker='o',mfc='r',label='Test Range') #Create a single line for the label
ax.legend(loc=2,numpoints=2,prop={'size':16}) # How can I add a label with different symbols the line segments?

plt.show()
  

Конечный продукт должен иметь легенду с символом, показывающим линию, соединяющую круг и квадрат.

Боюсь, вам придется комбинировать разные патчи mpatches , я не уверен, есть ли лучшее решение

 import matplotlib.pyplot as plt
import numpy as np
import matplotlib.patches as mpatches
from matplotlib.legend_handler import HandlerPatch
from matplotlib.legend_handler import HandlerLine2D


class HandlerCircle(HandlerPatch):
    def create_artists(self,legend,orig_handle,
                       xdescent,ydescent,width,height,fontsize,trans):
        center = 0.5 * width, 0.5 * height
        p = mpatches.Circle(xy=center,radius=width*0.3) 
        self.update_prop(p,orig_handle,legend)
        p.set_transform(trans)
        return [p]
    
class HandlerRectangle(HandlerPatch):
    def create_artists(self, legend, orig_handle,
                       xdescent, ydescent, width, height, fontsize, trans):
        center = 0,height/2-width*0.5/2
        width,height = width*0.5,width*0.5
        p = mpatches.Rectangle(xy=center,width=width,height=width) 
        self.update_prop(p, orig_handle, legend)
        p.set_transform(trans)
        return [p]

fig,ax = plt.subplots(figsize=(12,8))
texts = ['','','Test Range']
line, = ax.plot([],[],c='k')
c = [mpatches.Circle((0.,0.,),facecolor='r',linewidth=.5),
     line,
     mpatches.Rectangle((0.,0.),5,5,facecolor='r',linewidth=.5)]

ax.legend(c,texts,bbox_to_anchor=(.25,.95),loc='center',ncol=3,prop={'size':20},
          columnspacing=-1,handletextpad=.6,
          handler_map={mpatches.Circle: HandlerCircle(), 
                       line: HandlerLine2D(numpoints=0,),
                       mpatches.Rectangle: HandlerRectangle()}).get_frame().set_facecolor('w')

plt.show()
  

запустив этот скрипт, вы получите

Если вы используете другой размер рисунка или другой размер легенды, настройки в моем скрипте выше могут быть не оптимальными. В этом случае вы можете настроить следующие параметры:

1. Центры и размеры Circle и Rectangle
2. columnspacing и handletextpad в ax.legend(...)