pcolor情节力量颜色限制

Question

我正在使用plt.pcolor（numpy.matrix）

制作一个情节

矩阵中的值是固定的，它们只能是0,1,2,3或4。

然而，当我制作情节时，颜色限制是错误的。

我在这张照片上显示颜色边距。

cmap = matplotlib.colors.ListedColormap(['blue','green','red','black','yellow'])
plt.pcolor(matrix, edgecolors='black', linewidths=1,cmap=cmap)

所以，我想强制matplotlib采取：

blue=0
green=1
red=2
black=3
yellow=4

我怎样才能做到这一点？

即使在矩阵中没有任何值3，但有一些4s我仍然希望保留颜色编码。

修改

我使用以下方法做了一点改进：

plt.clim(0,4)

但仍然没有100％确定。