Matplotlib保存imshow数组

Question

我想知道是否有可能保存imshow函数的数组？这是什么意思？

我有一个二维数组，它的值是唯一的。我希望看到它以颜色表示，所以我自然会使用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]范围内。