Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Exclude matched array elements

How can I exclude matched elements of one array from another?

Postgres code:

a1 := '{1, 2, 5, 15}'::int[];
a2 := '{1, 2, 3, 6, 7, 9, 15}'::int[];

a3 := a2 ??magic_operator?? a1;

In a3 I expect exactly '{3, 6, 7, 9}'

Final Result

My and lad2025 solutions works fine.

Solution with array_position() required PostgreSQL 9.5 and later, executes x3 faster.

like image 292
Dmitry Avatar asked Jan 15 '16 21:01

Dmitry


Video Answer


1 Answers

It looks like XOR between arrays:

WITH set1 AS
(
 SELECT * FROM unnest('{1, 2, 5, 15}'::int[])
), set2 AS
(
 SELECT * FROM unnest('{1, 2, 3, 6, 7, 9, 15}'::int[])
), xor AS
(
  (SELECT * FROM set1
   UNION 
   SELECT * FROM set2)
  EXCEPT
  (SELECT * FROM set1
   INTERSECT 
   SELECT * FROM set2)
)
SELECT array_agg(unnest ORDER BY unnest)
FROM xor

Output:

"{3,5,6,7,9}"

How it works:

  1. Unnest both arrays
  2. Calculate SUM
  3. Calculate INTERSECT
  4. From SUM - INTERSECT
  5. Combine to array

Alternatively you could use sum of both minus(except) operations:

(A+B) - (A^B)
<=>
(A-B) + (B-A)

Utilizing FULL JOIN:

WITH set1 AS
(
 SELECT *
FROM unnest('{1, 2, 5, 15}'::int[])
), set2 AS
(
 SELECT *
 FROM unnest('{1, 2, 3, 6, 7, 9, 15}'::int[])
)
SELECT array_agg(COALESCE(s1.unnest, s2.unnest) 
                 ORDER BY COALESCE(s1.unnest, s2.unnest))
FROM set1 s1
FULL JOIN set2 s2
  ON s1.unnest = s2.unnest
WHERE s1.unnest IS NULL
  OR s2.unnest IS NULL;

EDIT:

If you want only elements from second array that are not is first use simple EXCEPT:

SELECT array_agg(unnest ORDER BY unnest)
FROM (SELECT * FROM unnest('{1, 2, 3, 6, 7, 9, 15}'::int[])
      EXCEPT
      SELECT * FROM unnest('{1, 2, 5, 15}'::int[])) AS sub

Output:

"{3,6,7,9}"
like image 179
Lukasz Szozda Avatar answered Sep 28 '22 12:09

Lukasz Szozda