python 生成的图像直接作为图像流返还给前端（flask方法实现）

最新推荐文章于 2024-04-08 16:51:33 发布

原创最新推荐文章于 2024-04-08 16:51:33 发布 · 6k 阅读

25 ·

CC 4.0 BY-SA版权

flask 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一个使用Flask框架、Matplotlib和NumPy动态生成图像，并将其作为响应返回给前端的方法。通过定义路由，服务器可以接收请求，生成包含线性数据的图像，然后将图像转换为PNG格式的数据流，最后发送给客户端。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from PIL import Image
from flask import Flask
import PIL
import matplotlib.pyplot as plt
import numpy as np
import io

app = Flask(__name__)

# 定义路由
@app.route("/")
def index():
    # 数据准备
    x = np.arange(10)
    y = x

    # plt.plot(x,y)
    # canvas = plt.get_current_fig_manager().canvas
    # canvas.draw()
    fig = plt.figure()
    plt.plot(x, y)
    canvas = fig.canvas
    # 上面这段代码和上面注释掉的代码效果一样

    # 方法1
    buffer = io.BytesIO()
    canvas.print_png(buffer)
    data = buffer.getvalue()
    buffer.close()

    # 方法2
    # buf, size = canvas.print_to_buffer()
    # image = Image.frombuffer('RGBA', size, buf, 'raw', 'RGBA', 0, 1)
    # buffer = io.BytesIO()
    # image.save(buffer, 'PNG')
    # data = buffer.getvalue()
    # buffer.close()

    # 向前端返回图像
    res = app.make_response(data)
    res.headers["Content-Type"] = "image/png"
    return res


if __name__ == "__main__":
    app.run()