2017-02-10 1 views
2

범례에 레이블 대신 이미지를 사용하고 싶습니다.Matplotlib 범례 레이블을 이미지로 대체하십시오.

예를 들어, 나는이 선을 그릴과 전설을 보여

import matplotlib.pyplot as plt 
plt.plot([1,2],label="first_image") 
plt.plot([2,1],label="second_image") 
plt.legend() 
plot.show() 

What I have now

을하지만 이런 식으로 뭔가하고 싶은이 그

The result I need

Insert image in matplotlib legend, 의 복제본이 아닙니다. 내 문제는 "레이블을 이미지로 변경"입니다. 다른 하나는 "범례 기호를 이미지로 변경"

+0

가능한 복제 (http://stackoverflow.com/questions/26029592/insert-image-in-matplotlib-legend) – Chuck

+1

하지 직접 관련이 있지만, 이 예제는 올바른 방향으로 인도 할 수 있습니다. http://matplotlib.org/examples/pylab_examples/demo_annotation_box.html – tom

답변

2

범례에서 이미지를 만드는 개념은 이미이 질문에 표시되어 있습니다 (Insert image in matplotlib legend). 이미지는 범례 항목의 아티스트로 사용됩니다.

범례에 선 핸들과 이미지가 필요한 경우, 서로 배치 된 두 핸들로 구성된 핸들을 만드는 것이 좋습니다. 유일한 문제는 매개 변수를 미세 조정하여 매력적으로 보입니다.

import matplotlib.pyplot as plt 
import matplotlib.lines 
from matplotlib.transforms import Bbox, TransformedBbox 
from matplotlib.legend_handler import HandlerBase 
from matplotlib.image import BboxImage 

class HandlerLineImage(HandlerBase): 

    def __init__(self, path, space=15, offset = 10): 
     self.space=space 
     self.offset=offset 
     self.image_data = plt.imread(path)   
     super(HandlerLineImage, self).__init__() 

    def create_artists(self, legend, orig_handle, 
         xdescent, ydescent, width, height, fontsize, trans): 

     l = matplotlib.lines.Line2D([xdescent+self.offset,xdescent+(width-self.space)/3.+self.offset], 
            [ydescent+height/2., ydescent+height/2.]) 
     l.update_from(orig_handle) 
     l.set_transform(trans) 

     bb = Bbox.from_bounds(xdescent +(width+self.space)/3.+self.offset, 
           ydescent, 
           height*self.image_data.shape[1]/self.image_data.shape[0], 
           height) 

     tbb = TransformedBbox(bb, trans) 
     image = BboxImage(tbb) 
     image.set_data(self.image_data) 

     self.update_prop(image, orig_handle, legend) 
     return [l,image] 


plt.figure(figsize=(4.8,3.2)) 
line, = plt.plot([1,2],[1.5,3], color="#1f66e0", lw=1.3) 
line2, = plt.plot([1,2],[1,2], color="#efe400", lw=1.3) 
plt.ylabel("Flower power") 

plt.legend([line, line2], ["", ""], 
    handler_map={ line: HandlerLineImage("icon1.png"), line2: HandlerLineImage("icon2.png")}, 
    handlelength=2, labelspacing=0.0, fontsize=36, borderpad=0.15, loc=2, 
    handletextpad=0.2, borderaxespad=0.15) 

plt.show() 

enter image description here

[하기 matplotlib 전설에 이미지 삽입]의
+0

감사합니다. 이것은 매력처럼 작동합니다! – Drico