очертить один элемент тепловой карты моря

Question

Можно ли обвести контур вокруг одного элемента в тепловой карте морского зоопарка?

Например, я хочу поместить желтый контур вокруг верхнего левого элемента на этом рисунке:

import matplotlib.pyplot as plt
import seaborn as sb

matrix = [[1, 2], [3, 4]]
index_of_element_to_outline = [0, 0]

sb.heatmap(matrix)
# insert some code to outline the given element 
plt.show()

Answer 1

Добавление прямоугольника так же просто, как,

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

matrix = [[1, 2], [3, 4]]
index_of_element_to_outline = [0, 0]

ax = sns.heatmap(matrix)
rect = plt.Rectangle(index_of_element_to_outline, 1,1, color="gold", 
                     linewidth=3, fill=False, clip_on=False)
ax.add_patch(rect)

plt.show()

Более сложное решение, в котором прямоугольник не перекрывает охватываемый пиксель, см. Установить край прямоугольника matplotlib за пределы указанной ширины? .