将数据从sqlalchemy移动到pandas DataFrame

Question

我正试着在一个熊猫数据目录中加载一个SQLAlchemy。

df = pd.DataFrame(LPRRank.query.all())

>>> df
0        <M. Misty || 1 || 18>
1        <P. Patch || 2 || 18>
...
...

0        M. Misty  1  18
1        P. Patch  2  18
...
...

当我尝试：

dff = pd.read_sql_query(LPRRank.query.all(), db.session())

我得到一个属性错误：

AttributeError: 'SignallingSession' object has no attribute 'cursor'

和

dff = pd.read_sql_query(LPRRank.query.all(), db.session)

AttributeError: 'scoped_session' object has no attribute 'cursor'

app = Flask(__name__)
db = SQLAlchemy(app)

class LPRRank(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    candid = db.Column(db.String(40), index=True, unique=False)
    rank = db.Column(db.Integer, index=True, unique=False) 
    user_id = db.Column(db.Integer, db.ForeignKey('lprvote.id'))

    def __repr__(self):
        return '<{} || {} || {}>'.format(self.candid,
                                                 self.rank, self.user_id) 

上一个问题SQLAlchemy ORM到pandas DataFrame的转换解决了我的问题，但是解决方案：使用query.session.bind不是我的解决方案。我使用db.session.add（）和db.session.commit（）打开/关闭会话，但是当我使用db.session.bind时，就会得到一个属性错误：

AttributeError: 'list' object has no attribute '_execute_on_connection'

Answer 1

只需在模型中添加__init__方法，并在构建dataframe之前调用Class对象。具体地说，下面创建一个元组的迭代绑定到使用pandas.dataframe()的列中。

class LPRRank(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    candid = db.Column(db.String(40), index=True, unique=False)
    rank = db.Column(db.Integer, index=True, unique=False) 
    user_id = db.Column(db.Integer, db.ForeignKey('lprvote.id'))

    def __init__(self, candid=None, rank=None, user_id=None):
        self.data = (candid, rank, user_id)

    def __repr__(self):
        return (self.candid, self.rank, self.user_id) 

data = db.session.query(LPRRank).all()
df = pd.DataFrame([(d.candid, d.rank, d.user_id) for d in data], 
                  columns=['candid', 'rank', 'user_id'])

或者，使用基于定义的模型类LPRRank的SQLAlchemy ORM来运行read_sql:

df = pd.read_sql(sql = db.session.query(LPRRank)\
                         .with_entities(LPRRank.candid,
                                        LPRRank.rank,
                                        LPRRank.user_id).statement, 
                 con = db.session.bind)