我尝试以这种方式使用 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(黄色) 颜色。哪里错了?
最佳答案
颜色图在 0 和 1 之间标准化。当它们用于绘图时,标准化会转移到数据的最小值和最大值。如果 0 和 1 是最小值和最大值,则 0 将是 map 的第一种颜色(红色),1 将是最后一种颜色(黄色)。
您需要的是将所需的颜色图行为考虑在内的规范化。最简单的选择是使用 vmin
和 vmax
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]])
plt.pcolormesh(xx, yy, zz, cmap = colors, vmin=0,vmax=colors.N)
关于python - PyPlot ColorMesh 使用错误的颜色映射创建绘图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46634600/