如何從給定的二維陣列製作彩色圖表？

Question

我一直在爲統一搜索，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圖形繪製的經驗。任何幫助感謝！

Answer 1

您可以使用matplotlib（或其他軟件包）。查看hist2d函數。 這裏有很多例子： http://matplotlib.org/gallery.html

其中一些你可能會覺得有用。