arrays - Matplotlib 保存 imshow 数组

Question

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

好吧，我有一个具有唯一值的二维数组。我想看到用颜色表示的，所以我自然会使用 imshow 函数。我知道 imshow 函数将颜色图应用于我的数组，然后显示它。我希望能够获得 matplotlib 用来显示我的原始二维数组的数组。这可以做到吗？

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是一个 shape 数组data.shape + (4,)，即 MxNx4，其中 4 是 (r,g,b,a)。您data需要缩放到范围内[0,1]。