从烧瓶中，如何将 matplotlib 绘图图像发送到 html？

Question

在flask中，如何让matplotlib中plt保存的图像显示在网页上？下面的 python 代码创建了一个绘图并将其保存为.png 图像。 HTML 文件也随之而来。当我运行它时，我得到了错误提示：

TypeError: expected str, bytes or os.PathLike object, not NoneType.

任何可以帮助我的人请告诉我该怎么做。

@app.route('/images/<air_quality>')
def images(air_quality):
    return render_template("displaygraph.html", title=air_quality)

@app.route('/fig/<air_quality>')
def fig(air_quality):
    import matplotlib.pyplot as plt
    import numpy as np
    from io import BytesIO

    img = BytesIO()
    pm10 = np.array(['23', '45', '56', '12'])
    pm25 = np.array(['34', '56', '59', '34'])
    dates = np.array(['2017-12-20', '2017-12-21', '2017-12-22', '2017-12-23'])

    plt.figure(figsize=(12, 8), dpi=100, facecolor='1.0')
    plt.title("GangNam", fontsize=20)
    plt.plot_date(dates, pm10, 'rs--', label='pm10')
    plt.plot_date(dates, pm25, 'gs--', label='pm25')
    plt.legend()
    img = plt.savefig('img.png')
    return send_file(img, mimetype='image/png')

我不在这里如何显示 HTML 文件。但是该文件在img 标签内有这个：

src="{{url_for('fig', air_quality = title)}}

Answer 1

您可以将 pylab 图形保存到内存文件中，例如 BytesIO(py3)、StringIO(py2)。

plt.savefig(img, format='png')
img.seek(0)

response=make_response(img.getvalue())
response.headers['Content-Type'] = 'image/png'
img.close()
return response