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

时间:2016-12-02 10:04:12

标签: python matplotlib

我有多条曲线在一个参数中有所不同,我想在一个图中绘制。为了区分它们,我想使用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()

结果如下:

starting point

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

Result of set_xlim

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

Result of set_xlim

我似乎需要的是调整显示颜色条限制的适当方法,或者创建自己的颜色条作为可用颜色条的子集的方法。有没有办法实现这些事情之一?

1 个答案:

答案 0 :(得分:4)

您可以使用我在下面的代码中编写的return $customers = DB::table('customers') ->leftJoin('contacts', function($join) { $join->on('contacts.customer_id', '=', 'customers.id'); $join->on(DB::raw('LENGTH(contacts.email) > 4')); }) 函数截断色彩映射。它会从现有的色彩映射中创建一个新的truncate_colormap

请注意,您不需要按matplotlib.colors.LinearSegmentedColormap缩放Normalise实例,并且在创建maxColor列表和{{1}时需要使用此新的colormap实例}}

colors

enter image description here