Itens principais com funções de janela

Análise de Séries Temporais no PostgreSQL

Jasmin Ludolf

Content Developer, DataCamp

Descrição da tabela

Tabela temperatures_monthly
Campos:
- station_id - id da estação meteorológica
- year_month - ano e mês
- t_monthly_min - temperatura mínima mensal em Celsius
- t_monthly_max - temperatura máxima mensal em Celsius
- t_monthly_avg - temperatura média mensal em Celsius

Agregação tradicional

SELECT station_id, year_month, t_monthly_min
FROM temperatures_monthly AS tm
JOIN

(   
  SELECT
        station_id, min(t_monthly_min) AS t_monthly_min
    FROM temperatures_monthly
    WHERE year_month BETWEEN '2018-01-01' AND '2018-12-31'
    GROUP BY station_id
) as p
USING(station_id, t_monthly_min)
WHERE year_month BETWEEN '2018-01-01' AND '2018-12-31'
ORDER BY station_id, t_monthly_min;

Agregação tradicional

|station_id|year_month|t_monthly_min|
|----------|----------|-------------|
|         1|2018-12-01|          5.4|
|         2|2018-12-01|          6.6|
|         3|2018-01-01|          1.6|
|         3|2018-02-01|          1.6|
|         4|2018-02-01|        -19.0|
|         6|2018-01-01|          2.6|
|         8|2018-02-01|        -16.3|

Problemas:

A estação 5 não aparece; não teve dados em 2018
Duas linhas para a estação 3; duas linhas tiveram a mesma mínima 3
Mostra só o mês mais frio; e se quisermos os dois meses mais frios?

Usando row_number

Precisamos numerar cada linha por estação
ROW_NUMBER(): numera as linhas por partição, em ordem crescente

SELECT station_id, year_month, t_monthly_min,
    ROW_NUMBER() OVER (PARTITION BY station_id ORDER BY t_monthly_min) AS rank
FROM temperatures_monthly AS tm
WHERE year_month BETWEEN '2018-01-01' AND '2018-12-31'
ORDER BY station_id, t_monthly_min;

Numerando as linhas

|station_id|year_month|t_monthly_min|rank|
|----------|----------|-------------|----|
|         1|2018-12-01|          5.4|   1|
|         1|2018-02-01|          6.2|   2|
|         1|2018-01-01|          7.4|   3|
|         1|2018-11-01|          9.0|   4|
|         1|2018-03-01|         10.6|   5| 
|         1|2018-04-01|         14.7|   6|
|         1|2018-10-01|         16.6|   7|
|         1|2018-05-01|         17.2|   8|
|         1|2018-06-01|         21.9|   9|
|         1|2018-09-01|         24.4|  10|
|         1|2018-07-01|         28.1|  11|
|         1|2018-08-01|         28.1|  12|
|         2|2018-12-01|          6.6|   1|
|         2|2018-01-01|          7.6|   2|
|         2|2018-02-01|          8.1|   3|

A numeração reinicia na estação 2
Porque particionamos por station_id

Temperaturas mais baixas

SELECT * FROM
(
    SELECT station_id, year_month, t_monthly_min,
    ROW_NUMBER() OVER 
          (PARTITION BY station_id ORDER BY t_monthly_min) AS rank
    FROM temperatures_monthly
    WHERE year_month BETWEEN '2018-01-01' AND '2018-12-31'
) AS q
WHERE rank < 3
ORDER BY station_id, rank;

Temperaturas mais baixas

|station_id|year_month|t_monthly_min|rank|
|----------|----------|-------------|----|
|         1|2018-12-01|          5.4|   1|
|         1|2018-02-01|          6.2|   2|
|         2|2018-12-01|          6.6|   1|
|         2|2018-01-01|          7.6|   2|
|         3|2018-01-01|          1.6|   1|
|         3|2018-02-01|          1.6|   2|

Valores mais altos

Para inverter a ordem, adicione DESC:

ROW_NUMBER() OVER (PARTITION BY station_id ORDER BY t_monthly_min DESC)

Vamos praticar!

Análise de Séries Temporais no PostgreSQL