plt.savefig() 部分裁剪颜色条标签中的下标字符，尽管使用 bbox_inches="tight"

Question

我有一个图形，其中在主图的 x 轴下方包含一个带标签的颜色条。当我尝试使用 plt.savefig() 保存它时，标签中下标字符的最底部会从保存的图像中裁剪掉，例如 ，尽管使用了 bbox_inches="tight"。但是，如果我只是在弹出窗口中手动保存图形，则下标字符不会被裁剪，例如。

虽然后一个图像可以手动裁剪，或使用代码中的附加行进行裁剪，但我将不胜感激有关如何解决此问题的任何建议，而无需进行额外的工作。

我尝试在颜色条标签中添加换行符，如下所示：

label="$U/U_{"+(u"\u221e")+"}$\n"

但这只是在标签下方添加了空白；下标字符的底部仍然被裁剪。

我也尝试添加行：

cb.set_label(label,labelpad=5)

但这只是将标签从颜色条底部偏移；标签下方没有提供额外的填充来完全显示下标字符。

代码如下：

import numpy
import random
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.patches as patches
import matplotlib.colors as mcolors
import matplotlib.colorbar as cbar
from matplotlib import cm

##########################################################
#  Centre colourmap so 0=white
class MidpointNormalize(mpl.colors.Normalize):
    def __init__(self,vmin=None,vmax=None,midpoint=None,clip=False):
        self.midpoint=midpoint
        mpl.colors.Normalize.__init__(self,vmin,vmax,clip)
    def __call__(self,value,clip=None):
        x,y=[self.vmin,self.midpoint,self.vmax],[0,0.5,1]
        return numpy.ma.masked_array(numpy.interp(value,x,y),numpy.isnan(value))
##########################################################

# Set min and max values
xymin=0
xymax=10
valmin=-5
valmax=5
val=numpy.zeros((xymax,xymax),dtype=float)

# Configure plot
fig,ax=plt.subplots()
ax.set_xlim([xymin,xymax])
ax.set_ylim([xymin,xymax])

# Configure colour bar
colours=plt.cm.RdBu(numpy.linspace(0,1,256)) 
colourmap=mcolors.LinearSegmentedColormap.from_list('colourmap',colours)
normalisecolors=mpl.colors.Normalize(vmin=valmin,vmax=valmax)
scalecolors=cm.ScalarMappable(norm=normalisecolors,cmap=colourmap)
label="$U/U_{"+(u"\u221e")+"}$"

for ix in range(xymin,xymax):
    for iy in range(xymin,xymax):
        xlow=ix*+1          # Calculate vertices of patch
        xhigh=(ix*1)+1
        ylow=iy*1
        yhigh=(iy*1)+1
        val[ix][iy]=random.randint(valmin,valmax)   # Patch value
        rgbacolor=scalecolors.to_rgba(val[ix][iy])  # Calculate RGBA colour for value
        ax.add_patch(patches.Polygon([(xlow,ylow),(xlow,yhigh),(xhigh,yhigh),(xhigh,ylow)],fill=True,facecolor=rgbacolor))  # Add value as polygon patch

cax,_=cbar.make_axes(ax,orientation="horizontal") 
cb=cbar.ColorbarBase(cax,cmap=colourmap,norm=MidpointNormalize(midpoint=0,vmin=valmin,vmax=valmax),orientation="horizontal",label=label)
plt.savefig("C:/Users/Christopher/Desktop/test.png",dpi=1200,bbox_inches="tight")
plt.clf
plt.close()

Answer 1

恐怕我真的没有一个好的答案给你。这似乎与此错误有关https://github.com/matplotlib/matplotlib/issues/15313

好消息是它正在处理中，坏消息是目前还没有修复。

无论如何要考虑两点（基于阅读 github 上的线程）：

• dpi越高，效果最差。所以你可能想以较低的 dpi 保存（300 对我来说很好）
• 问题不存在于 pdf 后端，因此您可以将绘图保存为 pdf（如果需要，最终转换为 png）

顺便说一句（这与有问题的错误无关）：我对您代码的复杂性感到困惑。在我看来，以下代码会产生相同的输出：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import TwoSlopeNorm

N=10
valmin=-5
valmax=5
valmid=0
val=np.random.randint(low=valmin, high=valmax, size=(N,N))
cmap = 'RdBu'
norm = TwoSlopeNorm(vcenter=valmid, vmin=valmin, vmax=valmax)
label="$U/U_{"+(u"\u221e")+"}$"

# Configure plot
fig, ax=plt.subplots()
im = ax.imshow(val, cmap=cmap, norm=norm, aspect='auto', origin='lower')
cbar = fig.colorbar(im, orientation='horizontal', label=label)
fig.savefig('./test-1200.png',dpi=1200,bbox_inches="tight")  # subscript is cut
fig.savefig('./test-300.png',dpi=300,bbox_inches="tight")  # subscript is not cut
fig.savefig('./test-pdf.pdf',dpi=1200,bbox_inches="tight")  # subscript is not cut

1200 dpi：

300 dpi：

pdf:

Answer 2

可能有点晚了，但我自己注意到，通过使 Figsize 更宽，我可以解决水平轴上的相同问题（右侧的颜色条标签仅在保存时被切断）。

以我为例

fig, ax = plt.subplots(1, 1, figsize=(20, 12), )

完成了这项工作，并将颜色条标签包含在保存的 png 图中。