我尝试以这种方式使用 xy 坐标和颜色图创建颜色网格图：

from matplotlib.colors import ListedColormap
import numpy as np
%pylab inline

colors = ListedColormap(['red', 'blue', 'yellow'])
xx,yy = np.meshgrid(np.arange(1, 6, 1), np.arange(1, 6, 1))
zz = np.array([[1,1,1,1,1],
               [1,1,1,1,1],
               [1,1,1,1,1],
               [2,2,0,0,0],
               [2,2,0,0,0]])
pyplot.pcolormesh(xx, yy, zz, cmap = colors)

当有三种颜色的列表并且我尝试将 xy 点映射到所有颜色中时，它可以正常工作（如上面的代码所示）：

但是当有三种颜色的列表，而我尝试将点仅映射到其中两种颜色时，映射就会出错：

zz = np.array([[1,1,1,1,1],
               [1,1,1,1,1],
               [1,1,1,1,1],
               [0,0,0,0,0],
               [0,0,0,0,0]])
pyplot.pcolormesh(xx, yy, zz, cmap = colors)

它应该映射到颜色 0（红色）和 1（蓝色），但我得到的图却有 0（红色）和2（黄色）。错误在哪里？