matplotlib 행렬의 특정 점 강조 표시

Question

imshow()을 사용하여 matplotlib에 행렬을 만들었습니다. 버튼을 누르면 매트릭스의 특정 플롯 포인트가 강조 표시됩니다. 내가 선택하고 싶은 목록에 좌표 집합이 있습니다. 내 행렬은 또한 이진 행렬입니다.

Answer 1

당신이 정확히 무엇을 요구하는지 이해한다면 나는 이것을 imshow 매트릭스 상단에있는 RGBA 오버레이로 보내고, 알파 채널은 '하이라이트'하려는 점을 제외하고는 0으로 설정합니다. . 그런 다음 강조 표시를 켜거나 끄기 위해 오버레이의 가시성을 토글 할 수 있습니다.

from matplotlib.pyplot import * 
import numpy as np 

def highlight(): 
    m = np.random.randn(10,10) 
    highlight = m < 0 

    # RGBA overlay matrix 
    overlay = np.zeros((10,10,4)) 

    # we set the red channel to 1 
    overlay[...,0] = 1. 

    # and we set the alpha to our boolean matrix 'highlight' so that it is 
    # transparent except for highlighted pixels 
    overlay[...,3] = highlight 

    fig,ax = subplots(1,1,num='Press "h" to highlight pixels < 0') 

    im = ax.imshow(m,interpolation='nearest',cmap=cm.gray) 
    colorbar(im) 
    ax.hold(True) 
    h = ax.imshow(overlay,interpolation='nearest',visible=False) 

    def toggle_highlight(event): 
     # if the user pressed h, toggle the visibility of the overlay 
     if event.key == 'h': 
      h.set_visible(not h.get_visible()) 
      fig.canvas.draw() 

    # connect key events to the 'toggle_highlight' callback 
    fig.canvas.mpl_connect('key_release_event',toggle_highlight)