如何用Pandas实现PostgreSQL表基于(item,date)索引的增改逻辑？

直接用Pandas的to_sql方法没法满足你按(item,date)复合索引匹配来增改的需求，因为它的if_exists参数没有提供基于特定键的冲突处理逻辑。下面是具体实现方案：

步骤1：将DataFrame写入临时表 #

先把处理好的stock_df写入PostgreSQL的临时表，临时表是会话级的，执行完自动销毁，不会污染原表：

# 写入临时表，每次运行替换原有临时表
stock_df.to_sql(
    name='temp_stock',
    con=conn_postgres,
    schema='arpan',
    if_exists='replace',
    index=False
)

步骤2：用PostgreSQL的ON CONFLICT实现Upsert #

通过原生SQL语句，把临时表的数据同步到原表arpan.stock，指定(item,date)作为冲突检测的唯一键，匹配到就更新对应字段，没匹配到就插入新记录：

# 假设原表除了item、date外，还有stock_qty、warehouse等字段，按需调整
upsert_sql = """
INSERT INTO arpan.stock (item, date, stock_qty, warehouse)
SELECT item, date, stock_qty, warehouse FROM arpan.temp_stock
ON CONFLICT (item, date) DO UPDATE
SET 
    stock_qty = EXCLUDED.stock_qty,
    warehouse = EXCLUDED.warehouse;
"""

# 执行SQL，建议放在事务中保证原子性
with conn_postgres.begin():
    conn_postgres.execute(upsert_sql)

关键说明： #

• 必须确保原表arpan.stock的(item,date)是唯一约束或主键约束（仅普通复合索引不行），否则ON CONFLICT无法识别冲突。如果还没创建，可以执行以下SQL添加：

  ALTER TABLE arpan.stock ADD CONSTRAINT stock_item_date_unique UNIQUE (item, date);
  

• SQL里的EXCLUDED是PostgreSQL的特殊关键字，代表当前试图插入但冲突的那条记录，用它来获取临时表中的更新值。

为什么原来的代码不行？ #

Pandas的to_sql没有if_exists='update'这个合法参数（合法值只有fail、replace、append），即使你误写了这个参数，它的逻辑也不会按指定主键判断冲突，而是会对全列重复的记录做处理，完全不符合你的需求。

内容的提问来源于stack exchange，提问作者Arpan Ghimire