matplotlib 如何在imshow中绘制单元格边框

nfg76nw0  于 2023-06-06  发布在  其他
关注(0)|答案(1)|浏览(141)

我有一个小的2D矢量显示:

import matplotlib.pyplot as plt
import numpy as np

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

如下:

但现在我想标记一个特定的细胞,以便将读者的注意力集中在该细胞上。因为这是特别感兴趣的。。
因此,像这个单元格的边框这样的东西会很好:

有人知道如何用matplotlib以方便的方式归档此文件吗?

h43kikqp

h43kikqp1#

在要高亮显示的像素位置放置一个矩形。

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

相关问题