我想将色彩映射传递给matplotlib.imshow
以绘制NxN数组,其中每个单元格只能取值-1,0和1。
我希望我的colormap能够将一种颜色(以十六进制代码形式)映射到每个值。
我根本不知道如何定义不是两种颜色之间渐变的自定义色彩映射。
答案 0 :(得分:3)
提供了一个非常类似问题的解决方案here。总之,colors.ListedColormap(list-of-colours)
是您的朋友。
答案 1 :(得分:3)
找到了我的问题的答案,正如费尔南多所说:
import matplotlib
colorsList = [(tuple rgb color 1),(tuple rgb color 2),(tuple rgb color 3)]
CustomCmap = matplotlib.colors.ListedColormap(colorsList)