保存 matplotlib table 会产生大量空白

Saving matplotlib table creates a lot of whitespace

我正在使用 matplotlib 和 python 2.7 创建一些 table。当我保存 tables 时,即使 table 只有 1 - 2 行,图像还是方形的,当我将它们添加到自动-稍后生成PDF。 我如何使用代码的示例在这里...

import matplotlib.pyplot as plt

t_data = ((1,2), (3,4))
table = plt.table(cellText = t_data, colLabels = ('label 1', 'label 2'), loc='center')
plt.axis('off')
plt.grid('off')
plt.savefig('test.png')

这会生成这样的图像... You can see you can see the white space around it

奇怪地使用 plt.show() 在 GUI 中生成没有白色 space 的 table。

我试过使用各种形式的tight_layout=True,但没有成功,也尝试过使背景透明(它变得透明,但仍然存在)。

如有任何帮助,我们将不胜感激。

由于 tables 是在轴内创建的,因此最终绘图大小将取决于轴的大小。所以原则上一个解决方案是要么设置图形大小,要么先设置坐标轴大小,然后让 table 适应它。

import matplotlib.pyplot as plt

fig = plt.figure(figsize=(6,1))

t_data = ((1,2), (3,4))
table = plt.table(cellText = t_data, 
                  colLabels = ('label 1', 'label 2'),
                  rowLabels = ('row 1', 'row 2'),
                  loc='center')

plt.axis('off')
plt.grid('off')

plt.savefig(__file__+'test2.png', bbox_inches="tight" )
plt.show()

另一种解决方法是让table原样绘制,并在保存前找出table的边界框。这允许创建一个非常紧围绕 table.

的图像
import matplotlib.pyplot as plt
import matplotlib.transforms

t_data = ((1,2), (3,4))
table = plt.table(cellText = t_data, 
                  colLabels = ('label 1', 'label 2'),
                  rowLabels = ('row 1', 'row 2'),
                  loc='center')

plt.axis('off')
plt.grid('off')

#prepare for saving:
# draw canvas once
plt.gcf().canvas.draw()
# get bounding box of table
points = table.get_window_extent(plt.gcf()._cachedRenderer).get_points()
# add 10 pixel spacing
points[0,:] -= 10; points[1,:] += 10
# get new bounding box in inches
nbbox = matplotlib.transforms.Bbox.from_extents(points/plt.gcf().dpi)
# save and clip by new bounding box
plt.savefig(__file__+'test.png', bbox_inches=nbbox, )

plt.show()