如何将熊猫数据帧发送到flask_restful api？

编程问答 2022-08-16

问题描述

现在，我一次向flask_restful api发送一个项目，并将结果附加到数据框。

如果api可以占用整个数据帧而不是逐行处理，那么它会更快。

这可能吗？

这是我正在做的非常基本的模型。

此api会使用此人的名字，并返回一个句子：“您的名字是...” 

from flask import Flask
from flask_restful import Resource,Api,reqparse

app = Flask(__name__)
api = Api(app)

class my_api(Resource):


    def __init__(self):
        self.reqparse = reqparse.RequestParser()
        self.reqparse.add_argument('name',type = str,default='')
        
        super(my_api,self).__init__()

    def get(self):

        args = self.reqparse.parse_args()
        name =  args['name']
        
        return f'Your name is {name}'

api.add_resource(my_api,'/')

if __name__ == '__main__':
    app.run(debug=True)

现在，我将在数据框上使用api：

df = pd.DataFrame({'name':['John','Jim','Mary','Michael']})
print(df)

      name
0     John
1      Jim
2     Mary
3  Michael


# list to save api results
results = []

# send each name in the column to the api
for name in df['name']:

    # api call
    api_data = send name to api
    
    results.append(api_data)

# attach results to dataframe
df['results'] = results

因此，按照我实际的Flask API的设置方式，如果我获得完整的数据帧而不是逐行显示，它将更快。而且，数据框比上面的模型大。有更多的列和行。

有什么建议吗？

解决方法

您可以将数据框转换为JSON并将其发送到API。您必须配置您的API以接受解析JSON。这样，您可以一次发送全部数据。也不使用{_3}}

作为类名使用snake_case
flaskflask-restfulpandaspandaspython