Matplotlib svg作为字符串而不是文件

Question

我想使用Matplotlib和pyplot来生成一个在Django框架中使用的svg图像。截至目前，我已经生成了链接到页面的图像文件，但有没有办法直接将svg图像作为unicode字符串获取而无需写入文件系统？

Answer 1

尝试使用StringIO以避免将任何类文件对象写入磁盘。

import matplotlib.pyplot as plt
import StringIO
from matplotlib import numpy as np

x = np.arange(0,np.pi*3,.1)
y = np.sin(x)

fig = plt.figure()
plt.plot(x,y)

imgdata = StringIO.StringIO()
fig.savefig(imgdata, format='svg')
imgdata.seek(0)  # rewind the data

svg_dta = imgdata.buf  # this is svg data

file('test.htm', 'w').write(svg_dta)  # test it

Answer 2

这是python3版本

import matplotlib.pyplot as plt
import numpy as np
import io

f = io.BytesIO()
a = np.random.rand(10)
plt.bar(range(len(a)), a)
plt.savefig(f, format = "svg")

print(f.getvalue()) # svg string