matplotlib 尽管使用了bbox_inches=“tight”,但plt.savefig()仍从colorbar标签中部分裁剪下标字符

7ivaypg9  于 2023-10-24  发布在  其他
关注(0)|答案(2)|浏览(151)

我有一个图,它包含一个在主图的x轴下面的带标签的颜色条。当我试图使用plt.savefig()保存这个图时,尽管使用了bbox_inches=“tight”,但标签中下标字符的底部会从保存的图像中被裁剪,就像this一样。然而,如果我只是在弹出窗口中手动保存这个图,下标字符不会被裁剪,比如this
虽然后一个图像可以手动裁剪,或者在代码中使用额外的行裁剪,但我将非常感谢任何关于如何解决这个问题的建议,而不需要这些额外的工作。
我尝试在colourbar标签中添加一个换行符,如下所示:

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()
jc3wubiy

jc3wubiy1#

恐怕我真的没有一个好的答案给你。这似乎是有关这个错误https://github.com/matplotlib/matplotlib/issues/15313
好消息是,它正在工作,坏消息是,目前还没有修复。
无论如何,有两点需要考虑(基于阅读github上的线程):

  • dpi越高,就越糟糕。所以你可能想以较低的dpi保存(300对我来说很好)
  • 这个问题不存在于pdf后端,所以你可以保存你的图在pdf(并最终转换为png如果需要)

顺便说一句(这与问题中的bug无关):我对你代码的复杂性感到困惑。在我看来,下面的代码产生了相同的输出:

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:

6kkfgxo0

6kkfgxo02#

这可能有点晚,但我注意到我自己可以解决同样的问题,在水平轴(在右边的颜色栏标签被切断,只有在保存),使figsize更广泛。
在我的例子中,fig, ax = plt.subplots(1, 1, figsize=(20, 12), )完成了这项工作,并在保存的png图中包含了colorbar标签。

相关问题