我有一个参数不同的多条曲线,我想在一张图中绘制。为了区分它们,我想使用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()
Run Code Online (Sandbox Code Playgroud)
结果如下:
现在,我希望颜色条在10处停止。但是,如果我只是xlim通过添加line ax[0].set_xlim(0, maxColor)来调整子图的,则正确调整了有色部分,但周围的盒子却被弄乱了: …