Numpy Matrix into a image

Question

我有一個int值的2d數組，我想轉換成圖像。

2d數組是在1-3之間隨機生成的，考慮數組中相鄰的int，我想在圖像中將1,2,3轉換爲R，G，B，以便更好地看到發生器的結果是。

這樣做的最好方法是什麼？

Answer 1

我會使用matplotlib庫。只需使用plt.imshow或plt.pcolormesh（第二個對於離散值在技術上更好）。在這種情況下，默認的顏色貼圖非常接近RGB，但如果您願意，可以使用another colormap。例如：

import numpy as np 
import matplotlib.pyplot as plt 

# Creating random 1-3 data in a 2D array 
data = np.random.randint(1,4,[100,150]) 

plt.pcolormesh(a) 

我使用IPython和%matplotlib inline，你可能需要調用plt.show()得到它，如果你不使用IPython的畫。