python - 检索 Matplotlib 热图颜色

Question

我正在尝试检索 matplotlib 热图上每个单元格的颜色，该热图由 imshow() 函数生成，例如由下面的 magic_function 执行:

import matplotlib.pyplot as plt
import numpy as np
hm = plt.imshow(np.random.rand(10, 10))
color_matrix = hm.magic_function() #returns matrix containing the RGB/Hex values of each cell

Answer 1

您正在寻找通过 imshow 创建的图像所使用的颜色图。现在，您当然可以像其他答案所建议的那样，首先对颜色图如何进入图像进行逆向工程。这看起来很麻烦，而且通常甚至是不可能的。

因此给定一个 AxesImage(imshow 返回的对象)或任何其他 ScalarMappable，您可以通过以下方式获得正在使用的颜色图.cmap。由于数据值被规范化到 0..1 之间的范围，您需要规范化，您可以从 .norm 属性中获得。最后，您需要从 .get_array() 方法中获取的数据。

magic_function 因此是三个函数的链。

im = plt.imshow(np.random.rand(10, 10))
color_matrix = im.cmap(im.norm(im.get_array()))

color_matrix 现在是与图像中的像素对应的 (10, 10, 4) 形 RGBA 颜色数组。