matplotlib中轮廓和轮廓f之间的颜色差异

Question

我正在尝试制作一个使用轮廓和轮廓线的图，它们都使用相同的颜色图，绘制相同的数据。 但是，contourf仅用于以某种方式（使用屏蔽数组）绘制“重要”数据。 同时，轮廓线用于绘制所有数据。 目的是绘制一个可以获取所有信息的图表，但将注意力吸引到更重要的区域。

我几乎可以按照自己的意愿进行这项工作，但是我发现轮廓线的颜色与Contourf中填充轮廓的颜色略有不同。

我猜这是因为轮廓颜色实际上是轮廓颜色之间的一半（这很有意义，因为轮廓线定义为一个值，例如1、2等）和填充轮廓在1和2之间，即“值”为1.5等）。

我将颜色图定义为

cmap = cm.coolwarm
cnorm=clrs.Normalize(cmap,clip=False)
cmap.set_under(color=cmap(0.0),alpha=1.0)
cmap.set_over(color=cmap(1.0),alpha=1.0)

我用于轮廓和轮廓线f的轮廓线水平是

clevs = [-3.,-2.,-1.,1.,2.,3.]

等高线绘制为

cplot=map.contour(x,y,diff,clevs,\
                      cmap=cmap,\
                      norm=cnorm,\
                      extend='both')

填充的轮廓绘制为

cplot=map.contourf(x,y,true_mask,clevs,cmap=cmap,\
                       norm=cnorm,
                       extend='both')

是否有一种简单的方法来使轮廓线的颜色“匹配”填充轮廓的颜色，即1处的线是1-2个填充轮廓的颜色，2处的线是2的颜色-3填充轮廓，-1处的线具有-2--1填充轮廓等的颜色？

非常感谢您的帮助。

Answer 1

我认为这里一种可能的解决方案是为轮廓调用创建新的颜色图。 如果原始色彩图定义为

cmap = matplotlib.cm.coolwarm
cnorm=matplotlib.colors.Normalize(cmap,clip=False)
cmap.set_under(color=cmap(0.0),alpha=1.0)
cmap.set_over(color=cmap(1.0),alpha=1.0)

和等高线图绘制为

clevs = [-3.,-2.,-1.,1.,2.,3.]

然后可以通过创建新的颜色图

cw=matplotlib.cm.get_cmap('coolwarm',8*(len(clevs)+1))
cw_vals=cw(np.arange(8*(len(clevs)+1)))
new_cw_vals=np.zeros([len(clevs),cw_vals.shape[1]],dtype=np.float128)
new_cw_vals_t = np.float128(cw_vals[4::8,:])
new_cw_vals_b = np.float128(cw_vals[12::8,:])
for i in np.arange(new_cw_vals.shape[0]):
    if clevs[i] < 0.0:
        new_cw_vals[i,:]=np.float32(new_cw_vals_t[i,:])
    else:
        new_cw_vals[i,:]=np.float32(new_cw_vals_b[i,:])
newcmap = matplotlib.colors.LinearSegmentedColormap.from_list("newcw", new_cw_vals)
newcnorm=matplotlib.colors.Normalize(newcmap,clip=False)

我不得不进行调整，以考虑到我没有绘制0线。

然后在轮廓调用中使用

cplot=map.contour(x,y,diff,clevs,
                      cmap=newcmap,
                      norm=newcnorm,
                      vmin=vmin,vmax=vmax)

本质上，我创建的色图是点数的8倍，然后挑选出中点。 不过，它还不是很完美，颜色与Contourf的颜色有些许差异。 这可能归结为舍入差异（颜色图值似乎是float32）。 尽管它可以很容易地更改为其他值，但它还是特定于clevs中使用的值。