Matplotlib只保存没有空格的文本

时间:2016-03-24 01:56:05

标签: python matplotlib sympy

所以在python中我有以下代码,取自this answer

text

这会打开文本并将其导出到png文件中:

Image Export 1

但是这包括超出框架外的空白的空格。您将如何裁剪图像以仅导出文本,如边界框,如此?

enter image description here

1 个答案:

答案 0 :(得分:4)

以下不是一个完美的解决方案,但它有望为您提供有关如何进步的一些想法:

import matplotlib.pyplot as plt
import sympy

x = sympy.symbols('x')
y = 1 + sympy.sin(sympy.sqrt(x**2 + 2))
lat = sympy.latex(y)

fig = plt.figure()
renderer = fig.canvas.get_renderer()
t = plt.text(0.001, 0.001, r"$%s$" % lat, fontsize = 50)
wext = t.get_window_extent(renderer=renderer)

fig.set_size_inches(wext.width / 65, wext.height / 40, forward=True)
fig.patch.set_facecolor('white')
plt.axis('off')
plt.tight_layout()
plt.savefig('out.png', bbox_inches='tight', pad_inches=0)
plt.show()

您可以通过使用当前渲染器获取窗口范围来确定文本大小。然后还可以手动指定图形尺寸。我虽然不确定正确的方法来转换两者之间。请注意,我为图像添加了边框,以便您可以看到剩余填充量:

matplotlib output

作为解决此问题的可能方法,以下方法仅使用Python的PIL库在保存之前自动裁剪图像:

import io
from PIL import Image, ImageOps
import matplotlib.pyplot as plt
import sympy

x = sympy.symbols('x')
y = 5 /sympy.sqrt(1 + sympy.sin(sympy.sqrt(x**2 + 2))) 
lat = sympy.latex(y)

fig = plt.figure()
t = plt.text(0.001, 0.001, r"$%s$" % lat, fontsize = 50)

fig.patch.set_facecolor('white')
plt.axis('off')
plt.tight_layout()

with io.BytesIO() as png_buf:
    plt.savefig(png_buf, bbox_inches='tight', pad_inches=0)
    png_buf.seek(0)
    image = Image.open(png_buf)
    image.load()
    inverted_image = ImageOps.invert(image.convert("RGB"))
    cropped = image.crop(inverted_image.getbbox())
    cropped.save('out.png')