如何从给定的二维阵列制作颜色图?

时间:2016-02-19 22:06:56

标签: python matplotlib

我一直在为搜索树编码进行统一搜索,A *搜索和贪婪搜索,并且我需要制作搜索在找到目标时所采用的路径的二维图。

示例2-D数组如下所示:

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

1 个答案:

答案 0 :(得分:0)

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

其中一些您可能觉得有用。