Презентация метки Spectral (SPy)

Question

Я хотел бы поставить легенду ниже изображения, нарисованного модулем spectral. Интересно, есть ли встроенный способ сделать это? Я ничего не нашел о создании легенд в spectral API.

Вот пример

img = np.array([111, 112, 113, 121, 122, 123, 131, 132, 133, 
       211, 212, 213, 221, 222, 223, 231, 232, 233, 
       311, 312, 313, 321, 322, 323, 331, 332, 333]).reshape((3,3,3)) 
labels = np.array([0, 1, 1, 2, 2, 2, 3, 3, 3]).reshape((3,3)) 

Я могу нарисовать img с labels этом так:

from spectral import imshow as spyShow 
imageView = spyShow(data=img, bands=(0,1,2), classes=labels, fignum=1, interpolation='none') 
imageView.set_display_mode('overlay') 

Теперь я хотел бы поместить легенду под изображением.

labelDictionary={0:'Unknown', 1:'Gravel', 2:'Aslphalt', 3:'Glass'} 

Из исходного кода я вижу, что цвета этикетки взяты из:

spectral.spy_colors 

Далее они нарисованы с помощью следующего кода:

import matplotlib.pyplot as plt 
from matplotlib.colors import ListedColormap, NoNorm 
cm = ListedColormap(np.array(spectral.spy_colors)/255.) 
plt.imshow(a, cmap=cm, vmin = 0, interpolation='none', norm=NoNorm()) 

Я думаю, что я мог бы извлечь те цвета и сопоставить их с метками и именами меток, используя специально выполненную функцию. Это правильный способ сделать легенду, или есть готовый путь, ради не изобретать колесо ...

Answer 1

Я добавил легенду так:

from matplotlib import patches 
from spectral import spy_colors 
labelPatches = [ patches.Patch(color=spy_colors[x]/255., 
       label=labelDictionary[x]) for x in np.unique(labels) ] 

После того как вы заплатку этикетки легенда легко добавить:

plt.legend(handles=labelPatches, ncol=2, fontsize='medium', 
      loc='upper center', bbox_to_anchor=(0.5, -0.05));