Matplotlib保存imshow数组
我想知道是否可以保存imshow函数的数组？ 这是什么意思？
那么，我有一个具有唯一值的二维数组。 我希望看到用颜色表示，所以我自然会使用imshow函数。 我明白，imshow函数将颜色映射应用于我的数组，然后显示它。 我希望能够获得matplotlib用来显示我的原始2d数组的颜色。 这可以做到吗？
实际上你可以用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]范围内。