Matplotlib: 使用三种颜色自定义色图

6

我希望向matplotlib.imshow传递一种颜色映射,以绘制一个NxN的数组,其中每个单元格只能取值-1、0和1。

我希望我的颜色映射将每个值映射到一个确切的颜色(用十六进制代码表示)。

我不知道如何定义一种自定义颜色映射,该映射不是两种颜色之间的渐变。

2个回答

5

这里提供了一个非常相似的问题的解决方案,可以在此处查看。简而言之,

import matplotlib
matplotlib.colors.ListedColormap(list-of-colours)

是你的朋友。


3

按照fernando的建议,我找到了解决我的问题的答案:

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

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接