Пересечь по двум столбцам array_agg в одной строке - PullRequest
Новая
       15

Пересечь по двум столбцам array_agg в одной строке

3 голосов
/ 20 марта 2019

У меня есть простой набор данных Postgres, который выглядит следующим образом: 

INSERT INTO mytable (day, person)
values
('Monday', 'A'),
('Monday', 'B'),
('Tuesday', 'A'),
('Thursday', 'B');

Затем я запускаю запрос, который выдает два array_aggs следующим образом: 

SELECT *
FROM (select day as d1,
             array_agg(distinct person) as agg1
      from mytable
      group by day) AS AA
   cross join
     (select day as d2,
             array_agg(distinct person) as agg2
      from mytable
      group by day) AS BB

, который дает этот набор данных: 

Monday, {A,B}, Monday, {A,B}
Monday, {A,B}, Thursday, {B}
Monday, {A,B}, Tuesday, {A}
Thursday, {B}, Monday, {A,B}
Thursday, {B}, Thursday, {B}
Thursday, {B}, Tuesday, {A}
Tuesday, {A}, Monday, {A,B}
Tuesday, {A}, Thursday, {B}
Tuesday, {A}, Tuesday, {A}

Я хотел бы добавить пятый столбец к этому запросу, который определяет количество повторяющихся записей в agg1 и agg2 в каждой строке.

Так, например, первая строка будет 2, а вторая строка - 1. Я надеялся сделать это следующим образом, но это дает мне неоднозначную синтаксическую ошибку: 

SELECT *, count(select unnest(agg1) intersect select unnest(agg2))
FROM (select day as d1,
             array_agg(distinct person) as agg1
      from mytable
      group by day) AS AA
   cross join
     (select day as d2,
             array_agg(distinct person) as agg2
      from mytable
      group by day) AS BB

Ответы [ 2 ]

1 голос
/ 20 марта 2019

Postgresql имеет LATERAL.

Что может быть использовано для содержимого полей на уровне записи. 

create table mytable (day varchar(30), person varchar(1));

INSERT INTO mytable (day, person)
values
('Monday', 'A'),
('Monday', 'B'),
('Tuesday', 'A'),
('Thursday', 'B');

SELECT *
FROM (
  select day as d1,
             array_agg(distinct person) as agg1
      from mytable
      group by day) AS AA
   cross join
     (select day as d2,
             array_agg(distinct person) as agg2
      from mytable
      group by day
) AS BB
CROSS JOIN LATERAL 
(
   SELECT COUNT(*) AS MatchingPersons
   FROM
   (
     SELECT unnest(agg1) person
     INTERSECT
     SELECT unnest(agg2)
   ) q
) lat 

d1       | agg1  | d2       | agg2  | matchingpersons
:------- | :---- | :------- | :---- | --------------:
Monday   | {A,B} | Monday   | {A,B} |               2
Thursday | {B}   | Monday   | {A,B} |               1
Tuesday  | {A}   | Monday   | {A,B} |               1
Monday   | {A,B} | Thursday | {B}   |               1
Thursday | {B}   | Thursday | {B}   |               1
Tuesday  | {A}   | Thursday | {B}   |               0
Monday   | {A,B} | Tuesday  | {A}   |               1
Thursday | {B}   | Tuesday  | {A}   |               0
Tuesday  | {A}   | Tuesday  | {A}   |               1

дБ <> скрипка здесь

0 голосов
/ 20 марта 2019

Используя функцию из этого ответа , вы можете написать: 

SELECT *, array_length(array_intersect(arr1, arr2), 1) AS repeat_count
FROM /* your query */
...