使用psycopg2 copy_from向PostgreSQL插入数据时serial列报错

这个问题的核心是**copy_from默认会尝试将你的数据匹配目标表的所有列**，而你的userdata表最后还有一个id（serial类型）列，但你的DataFrame只提供了前3列数据，PostgreSQL自然会认为你缺少id列的数据，从而抛出错误。

解决方案：明确指定要插入的列 #

你只需要在copy_from方法中添加columns参数，明确告诉PostgreSQL你要填充哪些列（也就是你的DataFrame包含的userid、activity、time三列），这样PostgreSQL就会自动忽略id列，使用它的默认序列值来填充。

修改后的代码如下：

import io
import pandas as pd

# 假设你的df已经定义完成
output = io.StringIO()
df.to_csv(output, sep='|', index=False, header=False)
connection = engine.raw_connection()
cursor = connection.cursor()
output.seek(0)
# 关键修改：添加columns参数，指定要插入的列
cursor.copy_from(output, 'userdata', sep='|', null='', columns=('userid', 'activity', 'time'))
connection.commit()
cursor.close()
connection.close()

为什么这样有效？ #

当你指定columns参数后，copy_from会严格按照你给出的列名顺序，将数据对应到表中的这些列，而不会尝试匹配表的所有列。由于id列没有被包含在columns里，PostgreSQL会自动使用它的默认值（nextval('userdata_id_seq'::regclass)）来填充，完全符合你的需求。

另外需要注意：确保columns参数里的列名和你的DataFrame列名、表中的列名完全一致（PostgreSQL默认对标识符大小写不敏感，但如果你的列名是带引号的特殊格式，要注意精准匹配）。

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