调整pcolormesh颜色条以专注于特定的绘图区域?

时间:2014-05-12 18:49:15

标签: python matplotlib

我有一个带有相关颜色条的pcolormesh实例。让我们用关联值Z = f(X,Y)调用我的数据点(X,Y)。我的数据Z经历了相当大的范围,我想专注于我的(X,Y)空间中的特定区域,其中Z的这种变化要小得多。如果我只是放大图或更改xlim / ylim值,那么我的颜色栏保持不变,我无法区分我感兴趣的区域中发生的微小变化(我希望颜色条改变以反映这个较小的地区)。

显然,我可以简单地修剪我感兴趣的区域之外存在的所有点的数据集,然后绘制它。但我想知道是否有更好的方法来做到这一点而不会弄乱数据。

1 个答案:

答案 0 :(得分:1)

您可以在vmin中设置关键字参数vmaxpyplot.pcolormesh

from numpy import linspace,meshgrid,pi,cos,sin
from matplotlib.pyplot import figure, show

x=linspace(0,100);y=linspace(0,100)
X,Y=meshgrid(x,y)
Z=cos(3*pi*X)+sin(6*pi*Y)

fig=figure()
ax=fig.add_subplot(111)
pc=ax.pcolormesh(X,Y,Z,vmin=0.3,vmax=0.6)
fig.colorbar(pc)
show()

这会使超出范围的值饱和,因此它们都会显示为相同的颜色。快乐的matplotlibbing。