首页 | 编程 | python

Python matplotlib图例放在外侧保存时显示不完整问题解决

2020年07月28日 作者：Poul_henry原文

上次说到的，使用如下代码保存矢量图时，放在外侧的图例往往显示不完整：

import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x1 = np.random.uniform(-10, 10, size=20)
x2 = np.random.uniform(-10, 10, size=20)
#print(x1)
#print(x2)
number = []
x11 = []
x12 = []
for i in range(20):
  number.append(i+1)
  x11.append(i+1)
  x12.append(i+1)
plt.figure(1)
# you can specify the marker size two ways directly:
plt.plot(number, x1, 'bo', markersize=20,label='a') # blue circle with size 20
plt.plot(number, x2, 'ro', ms=10,label='b') # ms is just an alias for markersize
 
lgnd=plt.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0,numpoints=1,fontsize=10)
lgnd.legendHandles[0]._legmarker.set_markersize(16)
lgnd.legendHandles[1]._legmarker.set_markersize(10)
 
plt.show()
 
fig.savefig('scatter.png',dpi=600)

保存为scatter.png之后的效果为：

以上述图为例，现考虑既然图例右侧没有显示，则调整subplots_adjust()函数的right参数，使其位置稍往左移，将参数right默认的数值0.9改为0.8，那么可以得到一个完整的图例：

import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x1 = np.random.uniform(-10, 10, size=20)
x2 = np.random.uniform(-10, 10, size=20)
#print(x1)
#print(x2)
number = []
x11 = []
x12 = []
for i in range(20):
  number.append(i+1)
  x11.append(i+1)
  x12.append(i+1)
plt.figure(1)
# you can specify the marker size two ways directly:
plt.plot(number, x1, 'bo', markersize=20,label='a') # blue circle with size 20
plt.plot(number, x2, 'ro', ms=10,label='b') # ms is just an alias for markersize
 
lgnd=plt.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0,numpoints=1,fontsize=10)
lgnd.legendHandles[0]._legmarker.set_markersize(16)
lgnd.legendHandles[1]._legmarker.set_markersize(10)
 
fig.subplots_adjust(right=0.8)
 
plt.show()
 
fig.savefig('scatter1.png',dpi=600)

保存为scatter1.png之后和scatter.png的对比效果为： 

图例legend在其它位置同理。 

2. 利用函数savefig()

上个博客讲到，使用savefig()函数中的三个参数fname, dpi, format可用以保存矢量图，现用该函数中另一个参数bbox_inches使

未保存到图中的图例包含进来。

下图可以看到，bbox_inches的作用是调整图的bbox, 即bounding box(边界框)

可以看到，scatter1.png，即第1种方法的思想，是将图像的右侧边界向左移动，截取该图用以保存的bbox未变；而scatter2.png，即第2种方法的思想，是直接将截取该图用以保存的bbox扩大为整个图像，而将其全部包括。

注：savefig()还有两个参数需要说明

其中一个是pad_inches，它的作用是当前面的bbox_inches为'tight'时，调整图像和bbox之间的填充距离，这里不需要设置，只要选择默认值即可。

这里举个例子，如果在图像左侧再加一个文本框text，保存图像时希望该文本框包含在bbox中，则可以使用该参数bbox_extra_artists将text包含进去（实际使用中，即使未使用bbox_extra_artists，保存的图像也包含该text）：

import numpy as np
import matplotlib.pyplot as plt
fig, ax = plt.subplots()
x1 = np.random.uniform(-10, 10, size=20)
x2 = np.random.uniform(-10, 10, size=20)
#print(x1)
#print(x2)
number = []
x11 = []
x12 = []
for i in range(20):
  number.append(i+1)
  x11.append(i+1)
  x12.append(i+1)
plt.figure(1)
# you can specify the marker size two ways directly:
plt.plot(number, x1, 'bo', markersize=20,label='a') # blue circle with size 20
plt.plot(number, x2, 'ro', ms=10,label='b') # ms is just an alias for markersize
 
lgnd=plt.legend(bbox_to_anchor=(1.05, 1), loc=2, borderaxespad=0,numpoints=1,fontsize=10)
lgnd.legendHandles[0]._legmarker.set_markersize(16)
lgnd.legendHandles[1]._legmarker.set_markersize(10)
 
text = ax.text(-0.3,1, "test", transform=ax.transAxes)
 
#fig.subplots_adjust(right=0.8)
plt.show()
fig.savefig('scatter3.png',dpi=600, bbox_extra_artists=(lgnd,text),bbox_inches='tight')

显示效果：

 为防止有的元素没有被包含在bbox中，可以考虑使用该参数

到此这篇关于Python matplotlib图例放在外侧保存时显示不完整问题解决的文章就介绍到这了,更多相关matplotlib外侧保存显示不完整内容请搜索来客网以前的文章或继续浏览下面的相关文章希望大家以后多多支持来客网！

相关链接