dataframe.to_sql 索引作为 postgresql 中的主键

Question

我有一个带有索引的数据框，我想将它存储在 postgresql 数据库中。为此我使用df.to_sql(table_name,engine,if_exists='replace', index=True,chunksize=10000)

pandas 数据框中的索引列被复制到数据库中，但未设置为主键。

有两种解决方案需要额外的步骤：

1. 指定架构df.to_sql(schema=) docs

2. 在表被摄取后设置主键。查询：

   ALTER TABLE table_name 添加主键（id_column_name）

有没有办法在不指定架构或更改表的情况下设置主键？

Answer 1

调用to_sql后：

import sqlalchemy
engine = create_engine()
engine.execute('ALTER TABLE schema.table ADD PRIMARY KEY (keycolumn);')

不幸的是，pandas.to_sql 没有设置主键，它甚至还破坏了现有表的主键。必须知道主键。