4

我正在尝试修改现有的一段 python 代码，该代码使用np.histogram2d. 我正在绘制其中的几个，我希望 y 轴和颜色范围在它们之间具有可比性。我找到了手动设置的方法y_limit，但现在我希望颜色范围也可以固定。下面的代码片段：

    hist,xedges,yedges = np.histogram2d(x,y, bins=[20, 50], range=[ [0, 100.0], [0, y_limit] ])
    # draw the plot
    extent = [xedges[0], xedges[-1], yedges[0], yedges[-1] ]
    im = ax.imshow(hist.T,extent=extent,interpolation='nearest',origin='lower', aspect='auto')
    # colormap, colorbar, labels, ect.
    im.set_cmap('gist_heat_r')
    divider = make_axes_locatable(ax)
    cax = divider.append_axes("right", "5%", pad="3%")
    ax.figure.colorbar(im, cax=cax)
    ax.set_title(d['name'] + ' foo')
    ax.set_xlabel("bar")
    ax.set_ylabel("coverage")

还有一个显示不同颜色范围的示例，其中一个从 0 到 2800 以上，另一个从 0 到 2400 以上。我希望能够将颜色范围设置为固定的最大值，例如两者都为 3000。有任何想法吗？

在此处输入图像描述 在此处输入图像描述

编辑：通过使用vmin和解决vmax。

4

1 回答 1

1

我从这篇文章中截取了一段代码

class MidpointNormalize(Normalize):
    def __init__(self, vmin=None, vmax=None, midpoint=None, clip=False):
    self.midpoint = midpoint
    Normalize.__init__(self, vmin, vmax, clip)

    def __call__(self, value, clip=None):
    # I'm ignoring masked values and all kinds of edge cases to make a
    # simple example...
    x, y = [self.vmin, self.midpoint, self.vmax], [0, 0.5, 1]
    return np.ma.masked_array(np.interp(value, x, y))

我像这样使用它：

norm = MidpointNormalize(midpoint=0.5,vmin = 0, vmax = 1)
ax.scatter(x, y,c=z, cmap = cm, norm = norm)

imshow也有参数norm，它应该可以工作。

于 2015-07-03T09:03:20.843 回答