使用 SQLAlchemy 的 ORM 将 SQL 数据库表转换为 DataFrame

Question

在 Flask 应用程序中，我想将 SQL Lite 数据库表（此处为

User

）转换为 DataFrame。 我试过这个

df = pd.DataFrame(dict(r) for r in User.query.all())

但它返回错误

'User' object is not iterable

。

我对 SQLAlchemy 不是很熟悉。正确的说法是什么？

Answer 1

试试这个

from your_flask_app import db
from your_flask_app.models import User
import pandas as pd

users = User.query.all()

user_dicts = [user.__dict__ for user in users]

for user_dict in user_dicts:
    user_dict.pop('_sa_instance_state', None)

df = pd.DataFrame(user_dicts)

print(df.head())