python - 根据值向网格图添加边框

Question

我想知道你是否能帮我解决这个问题。我有一个由 0 和 1 组成的网格，如果它是 1，我想向绘图单元格区域添加边框颜色。我使用 imshow 生成根据值着色的网格，例如:

a = np.random.randint(2, size=(10,10))

im = plt.imshow(a, cmap='Blues', interpolation='none', vmin=0, vmax=1, aspect='equal')

但是，我找不到要为 imshow 中的每个网格单元更改的任何边框属性。我读过 add_patch 可以用来在某些点放置一个矩形来模拟使用轴值的边框，但是有没有比循环和逐单元应用更好的方法？

Answer 1

没有内置函数可以对 imshow 边缘进行部分着色。最简单的选择可能确实是为每个单元格绘制一个矩形。

import matplotlib.pyplot as plt
import numpy as np

a = np.random.randint(2, size=(10,10))

im = plt.imshow(a, cmap='Blues', interpolation='none', vmin=0, vmax=1, aspect='equal')

def rect(pos):
    r = plt.Rectangle(pos-0.5, 1,1, facecolor="none", edgecolor="k", linewidth=2)
    plt.gca().add_patch(r)

x,y = np.meshgrid(np.arange(a.shape[1]),np.arange(a.shape[0]))
m = np.c_[x[a.astype(bool)],y[a.astype(bool)]]
for pos in m:
    rect(pos)

plt.show()