python - 为 pcolormesh 获取正确的颜色

Question

我正在使用 pcolormesh 来显示我正在运行的某些算法的结果。我为一些 x_min、x_max 等创建了通常的网格，并定义了我的颜色图

h = (x_max - x_min) / 1000.
xx, yy = numpy.meshgrid(numpy.arange(x_min, x_max, h), numpy.arange(y_min, y_max, h))
colours = ("blue", "green", "red")
cmap = colors.ListedColormap(colours)

然后执行 plt.pcolormesh(xx, yy, Z, alpha=0.7, cmap=cmap)，其中 Z 是我预测的结果(它可以是任何 0、1 或 2值，没关系)。

说 Z = numpy.zeros(xx.shape) 我应该看到所有东西都是蓝色的，Z = numpy.ones(xx.shape) 我应该看到绿色的，并且Z = 2*numpy.ones(xx.shape) 我应该看到红色，或者我是这么想的。相反，我总是看到蓝色。如果我添加这些行:

  Z[0] = 0
  Z[1] = 1
  Z[2] = 2

一切正常。看起来好像结果没有所有可能的结果(0,1,2)那么它默认只使用第一种颜色，蓝色，即使结果全是2，我想要红色。

我如何强制它具有我想要的颜色，即蓝色代表 0，绿色代表 1，红色代表 2，在每种情况下？

Answer 1

您可以使用 clim 来修复您的颜色条。

plt.clim(0, 3)

这将强制 0 为蓝色，1 为绿色，2 为红色。