如何从给定的2D阵列制作彩色图？

Question

我一直在为统一搜索、a*-搜索和贪婪搜索编写搜索树，并要求我在找到目标时绘制搜索路径的二维图

示例二维阵列如下所示：

grid = array([
    [2,3,0,0,0,0,0,0,0,0,0,0],
    [2,3,3,0,0,1,1,1,0,1,1,0],
    [2,1,3,3,0,0,1,1,0,1,1,0],
    [2,1,1,3,3,0,0,0,0,1,1,0],
    [2,1,1,1,3,0,0,0,0,1,1,0],
    [2,1,1,1,1,0,0,0,0,1,1,0],
    [2,1,1,1,1,1,1,1,1,1,1,0],
    [2,1,1,1,1,1,1,1,1,1,1,0],
    [0,2,0,0,0,0,0,0,0,0,1,0],
    [0,0,2,2,2,2,2,2,2,2,1,0],
    [0,0,1,1,1,1,1,1,1,1,1,0],
    [0,0,0,0,0,0,0,0,0,0,0,0]])

我需要知道如何根据存储在特定块中的值为颜色图中的每个块着色，即：

if 0 -> white
if 1 -> black
if 2 -> blue
if 3 -> yellow

我没有python二维图形绘制的经验。感谢您的帮助

Answer 1

您可以使用matplotlib（或其他软件包）。查看hist2d函数。 这里有很多例子： http://matplotlib.org/gallery.html

其中一些你可能会发现很有用