如何在matplotlib.pyplot.imshow中標記單元格（繪制單元格邊框）

Question

我要顯示一個小的2D向量：

import matplotlib.pyplot as plt
import numpy as np

img = np.random.rand(4,10)
plt.imshow(img, cmap='Reds')

如下：

但是現在我想標記一個特定的單元格，以使讀者將注意力集中在該單元格上。 因為這是特別感興趣的...

因此，像這個單元格的邊框這樣的東西會很好：

有人知道如何以方便的方式使用matplotlib將此文件存檔嗎？

Answer 1

在要突出顯示的像素的位置放置一個矩形。

import matplotlib.pyplot as plt
import numpy as np

def highlight_cell(x,y, ax=None, **kwargs):
    rect = plt.Rectangle((x-.5, y-.5), 1,1, fill=False, **kwargs)
    ax = ax or plt.gca()
    ax.add_patch(rect)
    return rect

img = np.random.rand(4,10)
plt.imshow(img, cmap='Reds')

highlight_cell(2,1, color="limegreen", linewidth=3)

plt.show()