Matplotlib：三种颜色的自定义色彩映射

Question

我想将色彩映射传递给matplotlib.imshow以绘制NxN数组，其中每个单元格只能取值-1,0和1。 我希望我的colormap能够将一种颜色（以十六进制代码形式）映射到每个值。

我根本不知道如何定义不是两种颜色之间渐变的自定义色彩映射。

Answer 1

提供了一个非常类似问题的解决方案here。总之，colors.ListedColormap(list-of-colours)是您的朋友。

Answer 2

找到了我的问题的答案，正如费尔南多所说：

import matplotlib
colorsList = [(tuple rgb color 1),(tuple rgb color 2),(tuple rgb color 3)]
CustomCmap = matplotlib.colors.ListedColormap(colorsList)