matplotlib：如何获得显示颜色？

Question

我已经使用绘制了图像

import matplotlib.pyplot as plt
plt.imshow(image)

image是一个只有5个不同值的NxM numpy数组。 如何获得在imshow显示的图像中这5个值已映射到的RGB值的列表？

Answer 1

如果您事先不知道图像中的值是什么，可以使用np.unique查找所有唯一值，然后使用AxesImage返回的imshow norm和cmap属性

例如：

import numpy as np
import matplotlib.pyplot as plt

im = plt.imshow(image)
colours = im.cmap(im.norm(np.unique(image))

Answer 2

好吧，我发现基于一种方法做做这个的其他问题。

import matplotlib.pyplot as plt
art = plt.imshow(image)
colours = [art.cmap(art.norm(c)) for c in range(1,6)]

颜色包含我需要的颜色。