Python向PostgreSQL插入可变维度数组报错的解决方法

问题重现 #

执行Python插入PostgreSQL数据时，持续报错：

'multidimensional arrays must have array expressions with matching dimensions'

相关信息：

• 原建表语句：

CREATE TABLE IF NOT EXISTS "participants" (
    "id"        BIGSERIAL NOT NULL PRIMARY KEY,
    "participants"      VARCHAR ARRAY);

• 原插入脚本：

script = 'INSERT INTO participants (id, participants) VALUES (%s, %s)'

• 待插入记录：

record = [1, [['Tom'], ['Steve', 'Ron'], ['Cheryl', 'Rick', 'Sue']]]
record = [2, [['Jane', 'Frank', 'Lin']]]
record = [3, [['Jose', 'Rhonda'], ['Stephanie', 'Joe']]]
record = [4, [['Pedro', 'Sandra']]]

• 预期存储结果：

错误原因 #

1. 表结构不匹配：原表participants列定义为VARCHAR ARRAY（一维数组），但要存储的是二维不规则数组（子数组长度不一致），维度不兼容导致解析失败。
2. PostgreSQL允许存储不规则多维数组，但需要明确列的维度类型，同时依赖psycopg2正确转换Python嵌套列表为PostgreSQL数组格式。

解决方案 #

步骤1：修改表结构（关键） #

将participants列改为二维字符数组类型，适配待存储的数据结构：

CREATE TABLE IF NOT EXISTS "participants" (
    "id" BIGSERIAL NOT NULL PRIMARY KEY,
    "participants" VARCHAR[][] -- 明确指定二维数组类型
);

步骤2：正确的Python插入脚本 #

使用psycopg2的参数化查询，直接传递嵌套列表即可，psycopg2会自动处理不规则数组的格式转换：

import psycopg2
from psycopg2 import OperationalError

# 建立数据库连接
def connect_db(db_name, db_user, db_pwd, db_host, db_port):
    try:
        conn = psycopg2.connect(
            database=db_name,
            user=db_user,
            password=db_pwd,
            host=db_host,
            port=db_port
        )
        return conn
    except OperationalError as e:
        print(f"数据库连接失败: {e}")
        return None

# 执行插入逻辑
conn = connect_db("你的数据库名", "你的用户名", "你的密码", "localhost", "5432")
if conn:
    cursor = conn.cursor()
    insert_sql = 'INSERT INTO participants (id, participants) VALUES (%s, %s)'
    
    # 整理所有待插入记录
    records = [
        [1, [['Tom'], ['Steve', 'Ron'], ['Cheryl', 'Rick', 'Sue']]],
        [2, [['Jane', 'Frank', 'Lin']]],
        [3, [['Jose', 'Rhonda'], ['Stephanie', 'Joe']]],
        [4, [['Pedro', 'Sandra']]]
    ]
    
    # 逐个插入记录
    for rec in records:
        cursor.execute(insert_sql, rec)
    
    conn.commit()
    cursor.close()
    conn.close()
    print("数据插入完成")

验证结果 #

执行查询语句：

SELECT * FROM participants;

会得到与预期完全一致的存储结果，PostgreSQL会用{{...}}格式表示二维数组，子数组长度差异不会影响存储。

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