postgresql Aggregate arrays into a single array with union of all elements

user_2727875 0
Viewed 10629

I'm looking for a sql pattern for an aggregate function to aggregate arrays. If I have 2 rows:

|id  |     array     |
|----+---------------|
|1   |    [1,2,3,4]  |
|1   |    [5,6]      |
|1   |    [7,8]      |
|--------------------|

And I want to do the following:

select id, *aggregate_function*(array) from table group by id

I want the result to be:

|id   | *aggregate_function*  |
|-----+-----------------------|
|1    | [1,2,3,4,5,6,7,8]     |
|-----------------------------|

There is no native postgres function that does this aggregation. But perhaps there's sql pattern that can be used here?

aggregate-functions postgresql arrays sql
3 Answers

Something like this should work:

with mytable as
(
select 1 as id, array[1, 2, 3, 4] as myarray

union all

select 1 as id, array[5, 6] as myarray

union all 

select 1 as id, array[7, 8] as myarray
)

select
  id,
  array_agg(elements order by elements)
from mytable, unnest(myarray) as elements
group by id

There's some discussion about building the custom function here: Concatenate/merge array values during grouping/aggregation

user_35071270

You can unnest and then group by:

WITH x (id, arr) AS (
  VALUES 
    (1, ARRAY[1,2,3,4])
  , (1, ARRAY[5,6])
  , (1, ARRAY[7, 8])
)
SELECT id, ARRAY_AGG(splitup) 
FROM (
    SELECT 
        id
      , unnest(arr) splitup 
    FROM x) splitup
GROUP BY 1
user_25702610

One option is to create a custom aggregate:

CREATE FUNCTION array_union(a ANYARRAY, b ANYARRAY)
RETURNS ANYARRAY AS
$$
  SELECT array_agg(x)
  FROM (
    SELECT unnest(a) x
    UNION ALL
    SELECT unnest(b)
  ) AS u
$$ LANGUAGE SQL;

CREATE AGGREGATE array_union_agg(ANYARRAY) (
  SFUNC = pg_temp.array_union,
  STYPE = ANYARRAY,
  INITCOND = '{}'
);

You can then simply use:

WITH x(arr) AS (
  VALUES
    (ARRAY[1,2,3,4]),
    (ARRAY[5,6]),
    (ARRAY[7,8])
)
SELECT array_union_agg(arr)
FROM x;

The query results in:

 array_union_agg
-----------------
 {1,2,3,4,5,6,7,8}

The above version does array concatenation.

Making true union as if the arrays were sets can be achieved by replacing UNION ALL with UNION in the definition of array_union function.

user_11371870
Related