Matplotlib에서 imshow()에 범례를 추가하는 방법은 무엇입니까?

Matplotlib에서 imshow()에 범례를 추가하려면 다음 단계를 수행할 수 있습니다. -

• 그림 크기를 설정하고 서브플롯 사이 및 주변 여백을 조정합니다.
• numpy를 사용하여 임의의 데이터 생성
• 색상 맵을 초기화합니다.
• 샘플 데이터에서 고유한 데이터 포인트를 가져옵니다(2단계).
• 범례에 배치할 다른 레이블과 색상으로 각 색상을 플로팅합니다.
• 상자의 오른쪽 상단 모서리에 범례를 배치합니다.
• 그림을 표시하려면 show()를 사용하세요. 방법.

예시

import numpy as np
from matplotlib import pyplot as plt, cm

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

data = np.random.rand(3, 3)
cmap = cm.YlOrBr
unique_data = np.unique(data)
i = 0

for entry in unique_data:
   mycolor = cmap(entry * 255 / (max(unique_data) - min(unique_data)))
   plt.plot(0, 0, "-", color=mycolor, label="%d"%i)
   i += 1

plt.imshow(data, cmap=cmap)
plt.legend(loc="upper right", bbox_to_anchor=(1.25, 1.0))

plt.show()

출력