PostgreSQL如何向二维数组追加元素？

问题根源 #

你遇到的报错核心是维度不匹配：

• 原代码中ARRAY[return_array]会把二维数组包装成三维数组（相当于给整个二维数组额外套了一层数组结构）
• 而ARRAY[to_char(x,'fm00000'), data_arr[x]]是一维数组，三维与一维数组无法直接拼接，循环到第三次时维度冲突就会触发错误。

正确写法 #

要向二维数组追加一行（即一个一维数组），你需要把待追加的一维数组作为二维数组的单个元素，再和原数组进行拼接：

方法1：使用拼接运算符|| #

直接将目标一维数组包装成二维数组的一行，再与原数组拼接：

DECLARE 
    x int DEFAULT 0;
    data_arr varchar[];
    return_array varchar[][];
BEGIN
data_arr := string_to_array(IN_ARRAY, IN_DELIMITER, '');
    
FOR x in array_lower(data_arr, 1)..array_upper(data_arr, 1) LOOP
    -- 将待追加的一维数组包装为二维数组的一行，再完成拼接
    return_array := return_array || ARRAY[ARRAY[to_char(x,'fm00000')::varchar, data_arr[x]::varchar]];
END LOOP;
    
RETURN return_array;
END;

方法2：使用array_append()函数 #

二维数组本质是「元素为一维数组的一维数组」，因此可以直接用array_append向外层一维数组追加元素（即你的目标一维数组）：

DECLARE 
    x int DEFAULT 0;
    data_arr varchar[];
    return_array varchar[][];
BEGIN
data_arr := string_to_array(IN_ARRAY, IN_DELIMITER, '');
    
FOR x in array_lower(data_arr, 1)..array_upper(data_arr, 1) LOOP
    return_array := array_append(return_array, ARRAY[to_char(x,'fm00000')::varchar, data_arr[x]::varchar]);
END LOOP;
    
RETURN return_array;
END;

更简洁的实现（无需手动循环） #

PostgreSQL可以通过generate_subscripts或unnest WITH ORDINALITY直接生成数组元素的索引，一步构建目标二维数组，效率更高：

CREATE OR REPLACE FUNCTION array_with_index(IN_ARRAY text, IN_DELIMITER text)
RETURNS text[][] AS $$
SELECT ARRAY(
    SELECT ARRAY[to_char(i, 'fm00000'), val]
    FROM unnest(string_to_array(IN_ARRAY, IN_DELIMITER)) WITH ORDINALITY AS arr(val, i)
)
$$ LANGUAGE sql STABLE;

这个写法通过unnest(... WITH ORDINALITY)直接获取每个元素的索引与对应值，避免了手动循环，代码更简洁高效。

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