从 Flask 将 Numpy 数组作为 CSV 返回

Question

我在 Flask 中有一部分 API，目前在 Json 中返回一个 Numpy 数组，我需要提供以 CSV 而不是 Json 的形式返回的选项。

我成功完成此操作的唯一方法是使用 numpy.savetxt 将 Numpy 数组保存为 CSV，然后提供该文件。我发现我不能像这样How can I generate file on the fly and delete it after download? 留下文件，但是仍然感觉“笨拙”

有没有办法在不通过文件的情况下将 Numpy 数组作为 CSV 返回？

Answer 1

是的，你可以，

@app.route('/download')
def download():
    csv = convert_numpy_array_to_csv(your_numpy_array)
    response = make_response(csv)
    response.headers["Content-Disposition"] = "attachment; filename=array.csv"
    return response

您不必将 csv 保存到文件中。如果您无法避免创建文件，您可以在临时文件夹中创建文件（可以通过import tempfile;tempfile.gettempdir() 获得）。这样做会在每次重新启动系统时自动删除文件

Answer 2

好吧，如果你想使用 numpy.savetxt 函数，那么你可以使用 cStringIO：

from cStringIO import StringIO
output = StringIO()
numpy.savetxt(output, numpy_array)
csv_string = output.getvalue()

对于 python 3，您可以从 io 模块导入 StringIO 或 BytesIO。