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]范围内。