在matplotlib中设置Colorbar范围

Question

我有以下代码：

import matplotlib.pyplot as plt

cdict = {
  'red'  :  ( (0.0, 0.25, .25), (0.02, .59, .59), (1., 1., 1.)),
  'green':  ( (0.0, 0.0, 0.0), (0.02, .45, .45), (1., .97, .97)),
  'blue' :  ( (0.0, 1.0, 1.0), (0.02, .75, .75), (1., 0.45, 0.45))
}

cm = m.colors.LinearSegmentedColormap('my_colormap', cdict, 1024)

plt.clf()
plt.pcolor(X, Y, v, cmap=cm)
plt.loglog()
plt.xlabel('X Axis')
plt.ylabel('Y Axis')

plt.colorbar()
plt.show()

因此，这将使用指定的颜色映射在X和Y轴上生成一个值“v”的图形。X轴和Y轴是完美的，但是colormap在v的min和max之间展开。我想强制colormap在0和1之间。

我想用：

plt.axis(...)

设置轴的范围，但这只接受X和Y的最小值和最大值的参数，而不接受colormap。

编辑：

为了清楚起见，假设我有一个图，它的值范围（0。。。0.3），以及另一个值为（0.2。。。0.8分）。

在这两个图中，我希望颜色条的范围是（0。。。1） 是的。在两张图中，我希望使用上面的全部cdict，这一颜色范围是相同的（因此两张图中的0.25将是相同的颜色）。在第一个图形中，0.3到1.0之间的所有颜色都不会出现在图形中，而是会出现在旁边的colourbar键中。在另一种情况下，0到0.2之间、0.8到1之间的所有颜色都不会出现在图形中，而是出现在旁边的颜色栏中。

Answer 1

使用vmin和vmax强制颜色的范围。下面是一个例子：

import matplotlib as m
import matplotlib.pyplot as plt
import numpy as np

cdict = {
  'red'  :  ( (0.0, 0.25, .25), (0.02, .59, .59), (1., 1., 1.)),
  'green':  ( (0.0, 0.0, 0.0), (0.02, .45, .45), (1., .97, .97)),
  'blue' :  ( (0.0, 1.0, 1.0), (0.02, .75, .75), (1., 0.45, 0.45))
}

cm = m.colors.LinearSegmentedColormap('my_colormap', cdict, 1024)

x = np.arange(0, 10, .1)
y = np.arange(0, 10, .1)
X, Y = np.meshgrid(x,y)

data = 2*( np.sin(X) + np.sin(3*Y) )

def do_plot(n, f, title):
    #plt.clf()
    plt.subplot(1, 3, n)
    plt.pcolor(X, Y, f(data), cmap=cm, vmin=-4, vmax=4)
    plt.title(title)
    plt.colorbar()

plt.figure()
do_plot(1, lambda x:x, "all")
do_plot(2, lambda x:np.clip(x, -4, 0), "<0")
do_plot(3, lambda x:np.clip(x, 0, 4), ">0")
plt.show()

Answer 2

使用CLIM函数（相当于MATLAB中的CAXIS函数）：

plt.pcolor(X, Y, v, cmap=cm)
plt.clim(-4,4)  # identical to caxis([-4,4]) in MATLAB
plt.show()

Answer 3

不确定这是否是最优雅的解决方案（这是我使用的），但您可以将数据缩放到0到1之间的范围，然后修改颜色栏：

import matplotlib as mpl
...
ax, _ = mpl.colorbar.make_axes(plt.gca(), shrink=0.5)
cbar = mpl.colorbar.ColorbarBase(ax, cmap=cm,
                       norm=mpl.colors.Normalize(vmin=-0.5, vmax=1.5))
cbar.set_clim(-2.0, 2.0)

通过这两个不同的限制，您可以控制颜色栏的范围和图例。在本例中，只有-0.5到1.5之间的范围显示在条形图中，而colormap覆盖-2到2（因此这可能是您的数据范围，在缩放之前记录）。

因此，不缩放颜色映射，而是缩放数据并使颜色栏与之匹配。