Question

我有一个我想要显示颜色贴图的补丁集合。由于我在colormap之上进行了一些操作，因此我无法使用matplotlib.colorbar实例来定义它。至少不是我能说的;这样做会剥夺我对颜色的一些操作，这些操作会删除缺少数据的补丁：

cmap = matplotlib.cm.YlOrRd
colors = [cmap(n) if pd.notnull(n) else [1,1,1,1]
          for n in plt.Normalize(0, 1)([nullity for _, nullity in squares])]

# Now we draw.
for i, ((min_x, max_x, min_y, max_y), _) in enumerate(squares):
    square = shapely.geometry.Polygon([[min_x, min_y], [max_x, min_y],
                                      [max_x, max_y], [min_x, max_y]])
    ax0.add_patch(descartes.PolygonPatch(square, fc=colors[i], 
                  ec='white', alpha=1, zorder=4))

所以我定义了一个matplotlib.colorbar.ColorbarBase实例，它起作用了：

matplotlib.colorbar.ColorbarBase(ax1, cmap=cmap, orientation='vertical',
                                 norm=matplotlib.colors.Normalize(vmin=0, vmax=1))

导致例如：

我遇到的问题是我想减小这个颜色条的大小（具体来说，将它缩小到特定的垂直尺寸，比如500像素），但我没有看到任何明显的做法这个。如果我有colorbar个实例，我可以使用axis property arguments轻松调整，但ColorbarBase缺少这些。

供进一步参考：

The example my implementation is based on。
The source code in question（警告：冗长）。

Answer 1

尺寸和形状由轴定义。这是我所拥有的代码片段，我将2个绘图组合在一起，并在顶部独立添加一个颜色条。我玩了add_axes实例中的值，直到我得到一个适合我的大小：

 cax = fig.add_axes([0.125, 0.925, 0.775, 0.0725]) #has to be as a list - starts with x, y coordinates for start and then width and height in % of figure width
 norm = mpl.colors.Normalize(vmin = low_val, vmax = high_val)     
 mpl.colorbar.ColorbarBase(cax, cmap = self.cmap, norm = norm, orientation = 'horizontal')

Answer 2

这个问题可能有点老了，但是我发现了另一种解决方案，对于不愿意为ColorbarBase类手动创建颜色条轴的任何人都可以提供帮助。

下面的解决方案使用matplotlib.colorbar.make_axes类从给定的轴创建一个依赖的sub_axes。然后可以为ColorbarBase类提供sub_axes以便创建colorbar。

该代码源自here

中描述的matplotlib代码示例

以下是代码段：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
import matplotlib.colorbar as mcbar
from matplotlib import ticker
import matplotlib.colors as mcolors


# Make some illustrative fake data:

x = np.arange(0, np.pi, 0.1)
y = np.arange(0, 2 * np.pi, 0.1)
X, Y = np.meshgrid(x, y)
Z = np.cos(X) * np.sin(Y) * 10


colors = [(1, 0, 0), (0, 1, 0), (0, 0, 1)]  # R -> G -> B

n_bins = [3, 6, 10, 100]  # Discretizes the interpolation into bins

cmap_name = 'my_list'


fig, axs = plt.subplots(2, 2, figsize=(9, 7))
fig.subplots_adjust(left=0.02, bottom=0.06, right=0.95, top=0.94, wspace=0.05)
for n_bin, ax in zip(n_bins, axs.ravel()):
    # Create the colormap
    cm = LinearSegmentedColormap.from_list(cmap_name, colors, N=n_bin)
    # Fewer bins will result in "coarser" colomap interpolation
    im = ax.imshow(Z, interpolation='nearest', origin='lower', cmap=cm)
    ax.set_title("N bins: %s" % n_bin)

    cax, cbar_kwds = mcbar.make_axes(ax, location = 'right',
                              fraction=0.15, shrink=0.5, aspect=20)




    cbar = mcbar.ColorbarBase(cax, cmap=cm, 
                              norm=mcolors.Normalize(clip=False), 
                              alpha=None, 
                              values=None, 
                              boundaries=None, 
                        orientation='vertical', ticklocation='auto', extend='both', 
                        ticks=n_bins, 
                        format=ticker.FormatStrFormatter('%.2f'), 
                        drawedges=False, 
                        filled=True, 
                        extendfrac=None, 
                        extendrect=False, label='my label')




    if n_bin <= 10:
        cbar.locator = ticker.MaxNLocator(n_bin) 
        cbar.update_ticks()

    else:
        cbar.locator = ticker.MaxNLocator(5) 
        cbar.update_ticks()


fig.show()

设置matplotlib ColorbarBase对象

2 个答案: