PostgreSQL中如何通过位置数组提取目标数组的多个元素？

你碰到的这个问题很常见——PostgreSQL的数组下标语法不支持直接传入一个位置数组来批量提取元素，这就是为什么你执行array2[array_positions(array1,'1')]会报错ERROR: array subscript must have type integer。不过别担心，有几种简单的方法能实现你想要的效果：

方法一：用unnest拆分位置数组再聚合 #

这是最直观的方式，先把位置数组拆成单个位置值，逐个提取元素后再重新聚合成数组：

WITH sample_data AS (
  SELECT 
    array['hello', 'bye', 'hello'] AS array2,
    array_positions(array[1,2,1], 1) AS target_positions
)
SELECT array_agg(array2[pos]) AS extracted_array
FROM sample_data, unnest(target_positions) AS pos;

执行后就能得到{'hello','hello'}的结果。注意这里我把array_positions的第二个参数改成了1（整数类型），和array1的元素类型保持一致，避免不必要的隐式转换。

方法二：更紧凑的数组构造器写法 #

如果不想用CTE，也可以直接用数组构造器结合子查询来实现：

SELECT array(
  SELECT array2[i]
  FROM unnest(array_positions(array[1,2,1], 1)) AS i
) AS extracted_array
FROM (SELECT array['hello', 'bye', 'hello'] AS array2) AS t;

这种写法更简洁，直接在array()构造器里完成所有逻辑。

方法三：自定义通用函数（适合频繁使用场景） #

如果你的业务中经常需要做这种操作，可以自定义一个函数来封装逻辑，以后调用起来更方便：

CREATE OR REPLACE FUNCTION array_extract_elements(source_array anyarray, positions_array int[])
RETURNS anyarray AS $$
BEGIN
  RETURN array(SELECT source_array[pos] FROM unnest(positions_array) AS pos);
END;
$$ LANGUAGE plpgsql IMMUTABLE;

定义完成后，直接调用函数就能得到结果：

SELECT array_extract_elements(array['hello', 'bye', 'hello'], array_positions(array[1,2,1], 1));

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