我有一个图,它包含一个在主图的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()
2条答案
按热度按时间jc3wubiy1#
恐怕我真的没有一个好的答案给你。这似乎是有关这个错误https://github.com/matplotlib/matplotlib/issues/15313
好消息是,它正在工作,坏消息是,目前还没有修复。
无论如何,有两点需要考虑(基于阅读github上的线程):
顺便说一句(这与问题中的bug无关):我对你代码的复杂性感到困惑。在我看来,下面的代码产生了相同的输出:
1200 dpi:
300 dpi:
pdf:
6kkfgxo02#
这可能有点晚,但我注意到我自己可以解决同样的问题,在水平轴(在右边的颜色栏标签被切断,只有在保存),使figsize更广泛。
在我的例子中,
fig, ax = plt.subplots(1, 1, figsize=(20, 12), )
完成了这项工作,并在保存的png图中包含了colorbar标签。