Matplotlib保存imshow數組

Question

我想知道是否可以保存imshow函數的數組？這是什麼意思？

嗯，我有一個具有唯一值的2d數組。我希望看到用顏色表示，所以我自然會使用imshow函數。我明白，imshow函數將顏色映射應用於我的數組，然後顯示它。我希望能夠獲得matplotlib用來顯示我的原始2d數組的顏色。這可以做到嗎？

Answer 1

實際上可以只得到顏色馬平與出imshow

data_ = (data - np.min(data))/ (np.max(data) - np.min(data)) 
my_cmap = matplotlib.cm.get_cmap('gray') # or what ever color map you want 
color_array = my_cmap(data_) 

color_array與是形狀data.shape + (4,)的陣列，即MxNx4與4存在（R，G，B，A）。您的data需要縮放到[0,1]範圍內。