python - 从 pandas df 更新数据库中的现有行

Question

我有一个 PostgreSQL 数据库。 Pandas 有一个“to_sql”函数，可以将数据帧的记录写入数据库。但是我还没有找到任何关于在我完成数据框后如何使用 pandas 更新现有数据库行的文档。

目前我可以使用 pandas read_sql_table 将数据库表读入数据框。然后，我会根据需要处理数据。但是，我无法弄清楚如何将该数据帧写回数据库以更新原始行。

我不想覆盖整个表格。我只需要更新最初选择的行。

Answer 1

一种方法是利用 sqlalchemy“表类”和 session.merge(row)、session.commit():

这是一个例子:

for row in range(0, len(df)):
    row_data = table_class(column_1=df.ix[i]['column_name'],
                           column_2=df.ix[i]['column_name'],
                           ...
                           )
    session.merge(row_data)
    session.commit()