ROLLUP и CUBE

PostgreSQL: сводная статистика и оконные функции

Michel Semaan

Data Scientist

Итоги по группам

Медали Китая и России на летних Олимпийских играх 2008 года по видам наград

| Country | Medal  | Awards |
|---------|--------|--------|
| CHN     | Bronze | 57     |
| CHN     | Gold   | 74     |
| CHN     | Silver | 53     |
| CHN     | Total  | 184    |
| RUS     | Bronze | 56     |
| RUS     | Gold   | 43     |
| RUS     | Silver | 44     |
| RUS     | Total  | 143    |

Старый способ

SELECT
  Country, Medal, COUNT(*) AS Awards
FROM Summer_Medals
WHERE
  Year = 2008 AND Country IN ('CHN', 'RUS')
GROUP BY Country, Medal
ORDER BY Country ASC, Medal ASC

UNION ALL

SELECT
  Country, 'Total', COUNT(*) AS Awards
FROM Summer_Medals
WHERE
  Year = 2008 AND Country IN ('CHN', 'RUS')
GROUP BY Country, 2
ORDER BY Country ASC;

Знакомство с ROLLUP

SELECT
  Country, Medal, COUNT(*) AS Awards
FROM Summer_Medals
WHERE
  Year = 2008 AND Country IN ('CHN', 'RUS')
GROUP BY Country, ROLLUP(Medal)
ORDER BY Country ASC, Medal ASC;

ROLLUP — это подпункт GROUP BY, добавляющий строки с итогами по группам
GROUP BY Country, ROLLUP(Medal) подсчитывает итоги по Country и Medal, а затем только по Country, заполняя Medal значениями null

ROLLUP — запрос

SELECT
  Country, Medal, COUNT(*) AS Awards
FROM summer_medals
WHERE
  Year = 2008 AND Country IN ('CHN', 'RUS')
GROUP BY ROLLUP(Country, Medal)
ORDER BY Country ASC, Medal ASC;

ROLLUP работает иерархически: агрегация выполняется от крайнего левого столбца к крайнему правому
- ROLLUP(Country, Medal) включает итоги по Country
- ROLLUP(Medal, Country) включает итоги по Medal
- Оба варианта включают общий итог

ROLLUP — результат

| Country | Medal  | Awards |
|---------|--------|--------|
| CHN     | Bronze | 57     |
| CHN     | Gold   | 74     |
| CHN     | Silver | 53     |
| CHN     | null   | 184    |
| RUS     | Bronze | 56     |
| RUS     | Gold   | 43     |
| RUS     | Silver | 44     |
| RUS     | null   | 143    |
| null    | null   | 327    |

Итоги по группам содержат null; строка, где все значения null, — это общий итог
Итоги по Medal отсутствуют, так как указано ROLLUP(Country, Medal), а не ROLLUP(Medal, Country)

Знакомство с CUBE

SELECT
  Country, Medal, COUNT(*) AS Awards
FROM summer_medals
WHERE
  Year = 2008 AND Country IN ('CHN', 'RUS')
GROUP BY CUBE(Country, Medal)
ORDER BY Country ASC, Medal ASC;

CUBE — это нeiерархический аналог ROLLUP
Он генерирует все возможные агрегации по группам
- CUBE(Country, Medal) подсчитывает итоги по Country, по Medal и общий итог

CUBE — результат

| Country | Medal  | Awards |
|---------|--------|--------|
| CHN     | Bronze | 57     |
| CHN     | Gold   | 74     |
| CHN     | Silver | 53     |
| CHN     | null   | 184    |
| RUS     | Bronze | 56     |
| RUS     | Gold   | 43     |
| RUS     | Silver | 44     |
| RUS     | null   | 143    |
| null    | Bronze | 113    |
| null    | Gold   | 117    |
| null    | Silver | 97     |
| null    | null   | 327    |

Обратите внимание: итоги по Medal включены

ROLLUP vs CUBE

Source

| Year | Quarter | Sales |
|------|---------|-------|
| 2008 | Q1      | 12    |
| 2008 | Q2      | 15    |
| 2009 | Q1      | 21    |
| 2009 | Q2      | 27    |

Используйте ROLLUP для иерархических данных (например, частей даты), когда не нужны все возможные агрегации
Используйте CUBE, когда нужны все возможные агрегации по группам

ROLLUP(Year, Quarter)

| Year | Quarter | Sales |
|------|---------|-------|
| 2008 | null    | 27    |
| 2009 | null    | 48    |
| null | null    | 75    |

CUBE(Year, Quarter)

Строки выше + следующие

| Year | Quarter | Sales |
|------|---------|-------|
| null | Q1      | 33    |
| null | Q2      | 42    |

Давайте потренируемся!

PostgreSQL: сводная статистика и оконные функции