如何从flask视图函数返回DataFrame并返回到另一个python请求

Question

Python 1 应用程序正在调用 url 来接收 DataFrame。

ret = requests.post(url, json = myobj)

另一个 python 2 应用程序 Flask 视图函数应该返回 Dataframe 。

falsk 函数无法返回 Dataframe 怎么办。

Answer 1

您可以选择 JSON 作为数据格式。
使用

pandas.DataFrame.to_json

，数据帧将转换为 JSON，然后可以发送。
另请参阅

pandas.read_json

将 JSON 数据转换回数据帧。

from flask import (
    Flask
)
import io 
import pandas as pd
import requests 

app = Flask(__name__)

@app.route('/')
def index():
    data = {
        'Name': ['Gerald','Benn','Charlotte'],
        'Age': [12, 25, 31],
    }
    df = pd.DataFrame(data, columns= ['Name', 'Age'])
    return app.response_class(
        df.to_json(orient='records'), 
        mimetype='application/json'
    )


@app.route('/retrieve')
def retrieve():
    r = requests.get(
        'http://127.0.0.1:5000/', 
        headers={'Accept': 'application/json'}
    )
    r.raise_for_status()
    df = pd.read_json(io.StringIO(r.text), orient='records')
    print(df)
    return ''