Aggiungi bordo personalizzato a determinate celle in una trama matplotlib/seaborn

Question

In questo momento sto usando la mappa dei cluster di Seaborn per generare alcune mappe di calore in cluster - fin qui tutto bene.

Per un determinato caso d'uso, ho bisogno di disegnare bordi colorati attorno a celle specifiche. C'è un modo per farlo? O con pcolormesh in matplotlib o in altro modo?

Answer 1

È possibile eseguire questa operazione sovrascrivendo un Rectangle patch nella cella che si desidera evidenziare. Usando la trama esempio dal seaborn docs

import seaborn as sns 
import matplotlib.pyplot as plt 
sns.set() 
flights = sns.load_dataset("flights") 
flights = flights.pivot("month", "year", "passengers") 
g = sns.clustermap(flights) 

possiamo evidenziare una cella facendo

from matplotlib.patches import Rectangle 
ax = g.ax_heatmap 

ax.add_patch(Rectangle((3, 4), 1, 1, fill=False, edgecolor='blue', lw=3)) 
plt.show() 

Questo produrrà la trama con una cella evidenziata in questo modo:

Nota l'indicizzazione delle celle è a 0 in base all'origine in basso a sinistra.