Добавить пользовательскую границу в определенные ячейки на графике matplotlib/seaborn

Question

Сейчас я использую кластерную карту Seaborn для создания кластеризованных карт тепла - пока все хорошо.

Для определенного варианта использования мне нужно нарисовать цветные границы вокруг определенных ячеек. Есть ли способ сделать это? Или с pcolormesh в matplotlib, или любым другим способом?

Answer 1

Вы можете сделать это, перестроив Rectangle patch на ячейку, которую вы хотели бы выделить. Используя в качестве примера сюжет из 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) 

Мы можем выделить ячейку, делая

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() 

Это произведет сюжет с выделенной ячейки следующим образом:

Примечание индексация ячеек равна 0 на основании начала в левом нижнем углу.