выделяя легенду (выделяется цветная полоса)

Question

Цветовая полоса маскируется тепловой картой, когда я использую следующий код для построения тепловой карты "total_avail" на XY-сетке, определенной переменными "y_me sh" и "x_me sh", где 3 Перечисленные выше переменные numpy массивы 800x800 и для сохранения тепловой карты поверх карты Google. Есть ли способ, я могу сделать его выдающимся, скажем, добавив прямоугольник (границы другого цвета) вокруг цветовой полосы?

import branca.colormap as cm
import matplotlib.pyplot as plt
import folium
import numpy as np
#1. save heatmap as png
NumChan = 7
fig = plt.figure(frameon=False)
ax = fig.add_axes([0, 0, 1, 1])
ax.axis('off')
cmap = plt.cm.get_cmap('RdBu', NumChan+1)
colors = cmap(np.arange(cmap.N))
plt.pcolormesh(y_mesh, x_mesh, overall_avail, cmap=cmap)
FileName = 'MyFile.png'
plt.savefig(FileName)
plt.close()
#2. save heatmap on map
m1 = folium.Map([LatCen,LonCen], zoom_start=12, tiles='stamentoner')
merc = os.path.join(FileName)
img = folium.raster_layers.ImageOverlay(name='Mercator projection SW',image=merc,bounds [latmin, lonmin], [latmax, lonmax]],opacity=0.75,interactive=True,cross_origin=False,zindex=1)img.add_to(m1)
# legend for html
colorlisttuples = []
for color in colors:
    colorlisttuples.append(tuple(color))
    colormap = cm.LinearColormap(colors=colorlisttuples, index=np.arange(0, NumChan + 1), 
    vmin=-0.5, vmax=0.5+NumChan)
    colormap.caption = 'Number'
    colormap.add_to(m1)
filenm = "FileName.html"
m1.save(outfile=filenm)

Это файл PNG