postgresql - 如何使用 pandas 中的数据框为 postgres 设置表中的主键？

Question

我想将一个表存储到 postgres 8.4(之前编辑过:8.1)使用 Pandas .DataFrame.to_sql使用以下命令通过 sqlalchemy。

df.to_sql("table_name",engine)

我正在使用 postgres。如何在要添加到数据库的表中设置 PRIMARY KEY？我尝试在 df.to_sql 中使用选项索引，但这似乎不起作用。

Answer 1

如果你的 df 包含你想用作主键的“id”列，那么这样做:

df.to_sql('table_name', engine, if_exists="append", index=False)
with engine.connect() as con:
    con.execute('ALTER TABLE table_name ADD PRIMARY KEY (id);')