在 group by 子句中连接数组

Question

我们在将数组分组为单个数组时遇到了问题。 我们希望将两列中的值连接到一个数组中，并将这些数组聚合为多行。

给定以下输入：

| id | name | col_1 | col_2 |
| 1  |  a   |   1   |   2   |
| 2  |  a   |   3   |   4   |
| 4  |  b   |   7   |   8   |
| 3  |  b   |   5   |   6   |

我们想要以下输出：

| a | { 1, 2, 3, 4 } |
| b | { 5, 6, 7, 8 } |

元素的顺序很重要，应该与聚合行的 id 相关。

我们尝试了array_agg() 函数：

SELECT array_agg(ARRAY[col_1, col_2]) FROM mytable GROUP BY name;

很遗憾，此语句引发错误：

ERROR: could not find array type for data type character varying[]

似乎不可能使用array_agg() 合并group by 子句中的数组。

有什么想法吗？

Answer 1

UNION ALL

你可以先用UNION ALL“反击”：

SELECT name, array_agg(c) AS c_arr
FROM  (
   SELECT name, id, 1 AS rnk, col1 AS c FROM tbl
   UNION ALL
   SELECT name, id, 2, col2 FROM tbl
   ORDER  BY name, id, rnk
   ) sub
GROUP  BY 1;

适用于生成您稍后请求的值的顺序。 The manual:

聚合函数array_agg、json_agg、string_agg和xmlagg， 以及类似的用户定义的聚合函数，产生 有意义的不同结果值取决于 输入值。默认情况下未指定此顺序，但可以 通过在聚合调用中编写 ORDER BY 子句来控制，如 如第 4.2.7 节所示。或者，从 排序的子查询通常会起作用。

我的大胆强调。

LATERAL subquery 和 VALUES 表达式

LATERAL 需要 Postgres 9.3 或更高版本。

SELECT t.name, array_agg(c) AS c_arr
FROM  (SELECT * FROM tbl ORDER BY name, id) t
CROSS  JOIN LATERAL (VALUES (t.col1), (t.col2)) v(c)
GROUP  BY 1;

同样的结果。只需要一次通过桌子。

自定义聚合函数

或者您可以创建一个自定义聚合函数，就像这些相关答案中讨论的那样：

• Selecting data into a Postgres array
• Is there something like a zip() function in PostgreSQL that combines two arrays?

CREATE AGGREGATE array_agg_mult (anyarray)  (
    SFUNC     = array_cat
  , STYPE     = anyarray
  , INITCOND  = '{}'
);

那么你可以：

SELECT name, array_agg_mult(ARRAY[col1, col2] ORDER BY id) AS c_arr
FROM   tbl
GROUP  BY 1
ORDER  BY 1;

或者，通常更快，但不是标准 SQL：

SELECT name, array_agg_mult(ARRAY[col1, col2]) AS c_arr
FROM  (SELECT * FROM tbl ORDER BY name, id) t
GROUP  BY 1;

添加的ORDER BY id（可以附加到此类聚合函数）保证了您想要的结果：

a | {1,2,3,4}
b | {5,6,7,8}

或者您可能对此替代方案感兴趣：

SELECT name, array_agg_mult(ARRAY[ARRAY[col1, col2]] ORDER BY id) AS c_arr
FROM   tbl
GROUP  BY 1
ORDER  BY 1;

生成二维数组：

a | {{1,2},{3,4}}
b | {{5,6},{7,8}}

最后一个可以用 Postgres 9.5 或更高版本中的内置 array_agg() 替换（并且应该是，因为它更快！） - 增加了聚合数组的功能：

SELECT name, array_agg(ARRAY[col1, col2] ORDER BY id) AS c_arr
FROM   tbl
GROUP  BY 1
ORDER  BY 1;

同样的结果。 The manual:

输入数组连接成一维数组（输入 必须都具有相同的维度，并且不能为空或 null）

所以和我们自定义的聚合函数array_agg_mult()不完全一样；

Answer 2

select n, array_agg(c) as c
from (
    select n, unnest(array[c1, c2]) as c
    from t
) s
group by n

或者更简单

select
    n,
    array_agg(c1) || array_agg(c2) as c
from t
group by n

---

解决新的订购要求：

select n, array_agg(c order by id, o) as c
from (
    select
        id, n,
        unnest(array[c1, c2]) as c,
        unnest(array[1, 2]) as o
    from t
) s
group by n