具有相同颜色的散点图和等高线图

Question

是否可以强制plt.scatter进入与plt.contourf和plt.contour相同的颜色级别？ 例如，我的代码可以绘制如下图：

为了制作第一个子图，我使用

cs=m[0].scatter(xs,ys,c=obsData,cmap=plt.cm.jet)
m.colorbar(cs)

为了制作第二个子图，我使用

cs2=m[1].contourf(x,y,areaData,cmap=cs.cmap)

对于每个后续的子图，我使用

m[ind].contourf(x,y,areaData,cmap=cs.cmap,levels=cs2.levels

其中 areaData 在循环中重新计算。

我的问题是，如何强制第一个子图与其他子图具有相同的颜色？ 我正在寻找与levels=cs2.levels关键字参数等效的参数。

Answer 1

正如您在评论中指出的，您的散点数据和等高线数据没有直接关系，但您希望将它们显示在同一个颜色图上。

我建议设置一个包含两组数据的通用颜色范围。 由于obsData指的是散点和areaData到轮廓，我会设置

vmin,vmax = (fun(np.concatenate([obsData,areaData])) for fun in (np.min,np.max))

确定所收集数据集的跨度（显然，要推广到多个输入数据集）。 这些可以传递给scatter和contourf来设置颜色映射的限制：

cs = m[0].scatter(xs,ys,c=obsData,cmap=plt.cm.viridis,vmin=vmin,vmax=vmax)
cs2 = m[1].contourf(x,y,areaData,cmap=cs.cmap,vmin=vmin,vmax=vmax)

一些手动增加跨度可能是为了获得漂亮的结果。

请注意，我将颜色图更改为viridis 。 如果您真的想公平地表示您的数据，这应该是您的第一步。