带有自定义cmap颜色的matplotlib散点图不正确

2024-10-05 10:02:40 发布

您现在位置:Python中文网/ 问答频道 /正文

所以我按照一个模式为我的散点图创建一个自定义的颜色图和cbar。我创建了4个子图,每个子图覆盖了一个参数的不同范围,这个参数用来给点颜色。此参数的值范围从1e-10到1.0。在

我做以下工作:

cmap = plt.cm.jet
# extract all colors from the .jet map
cmaplist = [cmap(i) for i in range(cmap.N)]
# force the first color entry to be grey
cmaplist[0] = (.5,.5,.5,1.0)
# create the new map
cmap = cmap.from_list('Custom cmap', cmaplist, cmap.N)

# define the bins and normalize
bounds = np.linspace(-10,0,11)
norm   = mpl.colors.BoundaryNorm(bounds, cmap.N)
...
fig, ((ax1, ax2), (ax3, ax4)) = plt.subplots(2, 2, sharex='col', sharey='row', figsize=(20,20))
...
ax1.scatter(xcoord[rng1], ycoord[rng1], s=massFilt[rng1]/dotNorm,
                    c=np.ma.log10(Zfilt[rng1]), cmap=cmap)
ax2.scatter(xcoord[rng2],ycoord[rng2], s=massFilt[rng2]/dotNorm,
                    c=np.ma.log10(Zfilt[rng2]), cmap=cmap)
ax3.scatter(xcoord[rng3], ycoord[rng3], s=massFilt[rng3]/dotNorm,
                    c=np.ma.log10(Zfilt[rng3]), cmap=cmap)
ax4.scatter(xcoord[rng4], ycoord[rng4], s=massFilt[rng4]/dotNorm,
                    c=np.ma.log10(Zfilt[rng4]), cmap=cmap)

其中RNG定义为

^{pr2}$

所以第一个面板只包括那些小于2e-7的,并且所有的点都是正确的颜色。。。我已经验证了过滤器是否正常工作:例如,Zfilt[rng4]只包含Zfilt>;1e-2。。。但不知何故,我在rng2和rng3中看到了只应用于rng1的颜色!!??见附件。在

有什么想法吗?我做错什么了?在

All points in panel 2 (top right) ARE less than 1e-4 and should be green, blue or grey... NOT RED?!?

以下是Zfilt[rng2]的示例-

Z values rng2  [-4.23451696 -4.35974369 -5.18479833 -6.17304647 -4.48839191 -5.16774006
 -4.12047222 -6.11491263 -5.81392662 -4.6491248  -4.75038775 -5.06640103
 -4.20821705 -4.12556725 -4.58661378 -4.17023495 -5.40845781 -4.54981553
 -5.82830048 -4.11185471 -4.43155534 -6.17025186 -4.88154584 -5.00024704
 -4.13626926 -5.57797731 -6.0617742  -5.33182163 -5.44963247 -5.88409509
 -6.16903327 -5.70808154 -4.88578943 -4.00873256 -4.1457824  -4.45174817
 -5.43829583 -4.32470978 -4.11634754 -5.1141915  -5.13310282 -4.15469421
...

Tags: the颜色npcmapmascatterxcoordycoord
1条回答
网友
1楼 · 发布于 2024-10-05 10:02:40

scatter()不知道数据的假定范围,因此它将颜色映射应用于它看到的整个范围,即从所提供数组的最小值到最大值。由于您需要不同的行为(所有子批次的范围相同,而不管数据如何),您需要通过vminvmax参数显式指定范围:

ax1.scatter(xcoord[rng1], ycoord[rng1], s=massFilt[rng1]/dotNorm,
                c=np.ma.log10(Zfilt[rng1]), cmap=cmap,
                vmin=-10, vmax=0)
ax2.scatter(xcoord[rng2],ycoord[rng2], s=massFilt[rng2]/dotNorm,
                c=np.ma.log10(Zfilt[rng2]), cmap=cmap,
                vmin=-10, vmax=0)
# etc

相关问题 更多 >

    热门问题