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

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