使用 matshow 时 matplotlib 中的自定义颜色

Question

在使用 matplotlib 时，是否有一种简单的方法可以为给定矩阵的每个元素指示特定颜色。例如，假设我们想用以下三种特定颜色显示“x”：红色、黑色和白色：

但是，我发现的唯一选项是使用“cmap”，它不会直接为您提供“直接”指定颜色的选项。

fig = plt.figure()
ax = fig.add_subplot(111)
x= [[0,0,0,0,0,0],[0,0,0,0,0,0], [0,1,1,2,1,1], [0,0,0,0,0,1], [0,1,1,1,1,1]]    
cax = ax.matshow(x,cmap=plt.cm.gray_r )
plt.show()

我的问题：我应该如何更改代码以显示上述红/黑/白网格？ [例如，0 表示黑色，1 表示白色，2 表示红色]，一般来说，我们如何才能为更大的颜色列表做到这一点？比如 10-15 种颜色。

另外，如何为matix中的某个元素分配某种颜色？例如在上面，x[i][j] == 0 然后 color ='black' 或 x[i][j] == 2 然后 color ='red'

谢谢。

Answer 1

您可以创建自己的颜色图：

from matplotlib.colors import ListedColormap

cmap = ListedColormap(['k', 'w', 'r'])
cax = ax.matshow(x,cmap=cmap)

如果您想指定 10-15 种颜色，您可能会用完单字母颜色。在这种情况下，您可以指定 RGB 三元组（例如ListedColormap([[0, 0, 0], [1, 1, 1], [1, 0, 0]])）或其他各种color formats。或者，使用here 列出的预定义离散（“定性”）颜色图之一。

如果矩阵中的值不是连续的整数，您可以在绘图前对其进行转换。

import numpy as np

x = np.array([[0,0,0,0,0,0],[0,77,0,0,22,0], [0,1,1,2,1,1], [0,0,14,0,0,1], [0,1,1,1,1,1]])
u, i = np.unique(x, return_inverse=True)
y = i.reshape(x.shape)
# array([[0, 0, 0, 0, 0, 0],
#        [0, 5, 0, 0, 4, 0],
#        [0, 1, 1, 2, 1, 1],
#        [0, 0, 3, 0, 0, 1],
#        [0, 1, 1, 1, 1, 1]])