分别为散点图保存颜色条

Question

我已经有了颜色条的散点图，我保存为PNG图像。我需要情节是一定的figsize，但添加colorbar缩放原始情节。

import pylab as plt 

plt.figure(figsize=FIGSIZE) 
plt.scatter(X, Y, c=Z, s=marker_size, norm=LogNorm(), vmin=VMIN, vmax=VMAX, cmap=CMAP,rasterized=True,lw=0,) 
CB = plt.colorbar(ticks=TICKS, format=FORMAT) 

我怎么能保存原始图（与figsize的设置如上）和彩作为两个单独的图像？

Answer 1

显而易见的答案是“分开绘制颜色条”。您需要创建一个新的数字窗口并在那里绘制颜色条，以防止第一个数字被扭曲。小例子：

import matplotlib.pyplot as plt 
import numpy as np # only for dummy data 

X,Y = np.mgrid[-2:3,-2:3] 
Z = np.random.rand(*X.shape) 
FIGSIZE = (2,3) 

plt.figure(figsize=FIGSIZE) 
mpb = plt.pcolormesh(X,Y,Z,cmap='viridis') 
# plot the original without a colorbar 
plt.savefig('plot_nocbar.png') 

# plot a colorbar into the original to see distortion 
plt.colorbar() 
plt.savefig('plot_withcbar.png') 

# draw a new figure and replot the colorbar there 
fig,ax = plt.subplots(figsize=FIGSIZE) 
plt.colorbar(mpb,ax=ax) 
ax.remove() 
plt.savefig('plot_onlycbar.png') 

# save the same figure with some approximate autocropping 
plt.savefig('plot_onlycbar_tight.png',bbox_inches='tight') 

考虑制作了以下四个数字（点击正确查看）：

首先是人物的已保存版本，而不到一个呼叫colormap。这很好，这是你想要保存的。第二张图显示如果我们拨打colorbar而没有任何额外的烦恼会发生什么：它需要原始图中的一些空间，这是你想要防止的。

您必须使用plt.subplots打开一个新图形（和轴），并使用原始图形的大小。通过这种方式，您可以确定生成的彩条将与原始图形中的彩条相同。在上面的设置中，我让matplotlib确定了颜色条本身的大小;但之后我们需要删除会污染结果图的辅助轴。 （另一个选择是手动创建一个单独的坐标轴，并带有预期的颜色条大小，我怀疑这不是一个可行的行为方式。）

现在，您可以在第三个绘图时，删除的轴之后留下的空白空间在结果图中清晰可见（但颜色条的大小相当完美）。您可以在后期制作中手动剪下此空白区域，或者使用可以自动剪切彩色条图像的东西。

我还包括一个版本的情节，其中matplotlib本身收获大部分数字：bbox_inches='tight'关键字参数savefig正是这样做的。好处在于，生成的图像文件只包含色条（如上图所示），但生成的色条大小与原始大小略有不同。根据您的特定需求，您需要尝试使用可用的方法来提供最适合您的解决方案。