使用python pyplot绘制网格以分隔像素

Question

我绘制了一个尺寸为6x7的2d numpy数组。背景全为白色（或全为黑色，无关紧要），像素为黄色或红色，并且可以在场中的任何地方。 通常不清楚红色和黄色像素之间的边界在哪里：

因此，我将背景更改为白色，并尝试在其中添加一个结果集的网格：

如何添加像素之间的网格？这应该相当快，因为​​我要打印成千上万张图像。我可以节省的每一毫秒都是有益的。

这是我使用的代码：

    data = np.ones((6, 7, 3), dtype=np.uint8)
    data = data*255
    data[where it should be yellow] = [255, 255, 0]
    data[where it should be red] = [255, 0, 0]
    #plt.axis('off')
    plt.rc('grid', linestyle="-", color='black')
    plt.grid(True)
    img = plt.imshow(data)#, interpolation='nearest')
    #img.axes.get_xaxis().set_visible(False) 
    #img.axes.get_yaxis().set_visible(False)
    plt.show()
    plt.savefig("image%02d.png" % i, bbox_inches='tight', pad_inches = 0)