python - 如何在 matplotlib 中仅使用和绘制颜色条的一部分？

Question

我有多个曲线，它们在一个参数上不同，我想在一个图中绘制。为了区分它们，我想使用 matplotlib 的颜色条之一。为此，我根据所述参数生成了一个颜色列表。此外，我想添加一个颜色条来解释所使用的颜色。我可以轻松地做到这一切。现在的问题是，我只想使用可用颜色图的一部分，因为它变得太亮，因此在某个阈值以上几乎不可见。但是当我现在只在一个子范围内选择颜色时，我没有找到调整显示颜色条范围的方法。

这是我想要实现的(几乎)最小的例子:

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

gs = gridspec.GridSpec(2, 1,
                       height_ratios=[1, 4]
                       )
ax = [plt.subplot(g) for g in gs]

parameterToColorBy = np.linspace(5, 10, 6, dtype=float)

maxColor = 0.85
colors = [plt.get_cmap("inferno")(i)
          for i in np.linspace(0, maxColor, parameterToColorBy.shape[0])]

norm = mpl.colors.Normalize(parameterToColorBy[0],
                            parameterToColorBy[0]+
                            (parameterToColorBy[-1]-parameterToColorBy[0])/
                            maxColor)
cb = mpl.colorbar.ColorbarBase(ax[0],
                               cmap="inferno",
                               norm=norm,
                               ticks=parameterToColorBy,
                               orientation='horizontal')
ax[0].xaxis.set_ticks_position('top')

for p, c in zip(parameterToColorBy, colors):
    ax[1].plot(np.arange(2)/p, c=c)

plt.show()

结果看如下:

我现在希望颜色条停在 10。但是如果我只是调整 xlim通过添加行 ax[0].set_xlim(0, maxColor) 子图，有色部分调整正确，但是周围的框乱了:

或者，我找到了一个颜色条函数 set_clim .但这只会改变规范化，似乎并没有像我想要的那样工作。添加cb.set_clim(parameterToColorBy[0], parameterToColorBy[-1])导致颜色改变但轴不变:

我似乎需要的是一种适当的方法来调整显示的颜色条的限制，或者一种创建自己的颜色条作为可用颜色条的子集的方法。有什么方法可以实现其中之一吗？

Answer 1

您可以使用我在下面的代码中编写的 truncate_colormap 函数来截断颜色图。它根据现有颜色图创建一个新的 matplotlib.colors.LinearSegmentedColormap。

请注意，您不需要通过 maxColor 缩放 Normalise 实例，并且您需要在创建 colors< 时使用这个新的颜色图实例 列表和 colorbar。

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec
import matplotlib.colors as mcolors

gs = gridspec.GridSpec(2, 1,
                       height_ratios=[1, 4]
                       )
ax = [plt.subplot(g) for g in gs]

parameterToColorBy = np.linspace(5, 10, 6, dtype=float)

def truncate_colormap(cmap, minval=0.0, maxval=1.0, n=-1):
    if n == -1:
        n = cmap.N
    new_cmap = mcolors.LinearSegmentedColormap.from_list(
         'trunc({name},{a:.2f},{b:.2f})'.format(name=cmap.name, a=minval, b=maxval),
         cmap(np.linspace(minval, maxval, n)))
    return new_cmap

minColor = 0.00
maxColor = 0.85
inferno_t = truncate_colormap(plt.get_cmap("inferno"), minColor, maxColor)

colors = [inferno_t(i)
          for i in np.linspace(0, 1, parameterToColorBy.shape[0])]

norm = mpl.colors.Normalize(parameterToColorBy[0],
                            parameterToColorBy[-1])

cb = mpl.colorbar.ColorbarBase(ax[0],
                               cmap=inferno_t,
                               norm=norm,
                               ticks=parameterToColorBy,
                               orientation='horizontal')

ax[0].xaxis.set_ticks_position('top')

for p, c in zip(parameterToColorBy, colors):
    ax[1].plot(np.arange(2)/p, c=c)

plt.show()