Прямо сейчас я использую кластерную карту Seaborn для создания кластерных тепловых карт — пока все хорошо.
Для определенного варианта использования мне нужно нарисовать цветные границы вокруг определенных ячеек. Есть ли способ сделать это? Или с pcolormesh в matplotlib, или любым другим способом?
Вы можете сделать это, нанеся прямоугольный патч на ячейку, которую вы бы хочу выделить. На примере графика из документов Seaborn
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 на основе источника в левом нижнем углу.
