使用FLask和Matplotlib即时生成图像

时间:2018-11-12 18:35:26

标签: python matplotlib flask

我想绘制图像而不必将其保存在静态文件夹中。

我有一个用户Messa在此处发布的示例,该示例运行得很好:

import io
import random
from flask import Response
from matplotlib.backends.backend_agg import FigureCanvasAgg as FigureCanvas
from matplotlib.figure import Figure

@app.route('/plot.png')
def plot_png():
    fig = create_figure()
    output = io.BytesIO()
    FigureCanvas(fig).print_png(output)
    return Response(output.getvalue(), mimetype='image/png')

def create_figure():
    fig = Figure()
    axis = fig.add_subplot(1, 1, 1)
    xs = range(100)
    ys = [random.randint(1, 50) for x in xs]
    axis.plot(xs, ys)
    return fig

然后在HTML文件中:

<img src="/plot.png" alt="my plot">

结果是完美的。我可以实时获得图像。

现在我想使用一些这样的参数:

@app.route('/<x_values_source>/<y_values_source>/plot.png')
def plot_png(x_values_source, y_values_source):
    fig = create_figure(x_values_source, y_values_source)
    output = io.BytesIO()
    FigureCanvas(fig).print_png(output)
    return Response(output.getvalue(), mimetype='image/png')

def create_figure(x_values_source, y_values_source):
    x_values = function_that_returns_x_values_set(x_values_source)
    y_values = function_that_returns_y_values_set(y_values_source)
    fig = Figure()
    axis = fig.add_subplot(1, 1, 1)
    axis.plot(x_values, y_values)
    return fig

但是当然不行。我是Python的新手,但我仍然在学习很多东西。

谢谢!

1 个答案:

答案 0 :(得分:0)

要传入参数,最好不要在URI前面加上前缀,因为这可能会使Flask的路由解码感到困惑。相反,如果您将URL设置为

/plot.png?x=1,2,3,4&y=100,200,300,400

您可以通过request.args来获取参数。它们将以字符串形式出现,因此您需要分开并转换为整数(如果使用实数则为浮点数)

from flask import request

@app.route('/plot.png')
def plot():
    try:
        x_values = [int(x) for x in request.args.get('x','').split(',')]
        y_values = [int(y) for y in request.args.get('y','').split(',')]
    except:
        # malformed 

我有一个有效的示例here,该示例具有动态处理matplotlib的可靠方法。总体思路是

from flask import send_file

@app.route('/plot.png')
def plot_png():
    fig, ax = plt.subplots()
    draw(ax)
    return fig_response(fig)

def fig_response(fig):
    """Turn a matplotlib Figure into Flask response"""
    img_bytes = BytesIO()
    fig.savefig(img_bytes)
    img_bytes.seek(0)
    return send_file(img_bytes, mimetype='image/png')

您提供draw()的位置。不过,请看一下我的示例,因为您要添加一些缓存头。