如何从给定的二维数组制作颜色图

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 2-d图形绘图的经验。任何帮助表示赞赏！

Answer 1

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

其中一些您可能会发现有用。