pcolor plot force颜色限制

Question

我正在用plt.p颜色(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采取：

^{pr2}$

我怎么才能做到呢？在

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

编辑：

我用以下方法做了一个小小的改进：

plt.clim(0,4)

但是仍然不是100%可以。在