显示独特的独特颜色

Question

我有一个通过读取图像获得的2d numpy数组。数组的唯一值是0,1和2.我想绘制分别为值0,1和2显示红色，绿色和蓝色的唯一颜色的图像。

plt.imshow(data, cmap=colors.ListedColormap(['red'])

你会怎么做？

Answer 1

from matplotlib.colors import from_levels_and_colors
cmap, norm = from_levels_and_colors([0,1,2,3],['red','green','blue'])
plt.imshow(data, cmap=cmap, norm=norm)