如何在 postgres 上使用开始和结束日期和季度生成系列

Question

我有一个如下所示的表格，我想使用开始日期和结束日期将每一行的值平均分配到每个季度的 3 个月到开始日期和结束日期之间的所有季度（最后两列)。

我熟悉 Postgres 中的生成序列和间隔，但我很难得到我想要的。

我的表有一个 ID 列将行组合在一起，一个 quarter 列指示 ID 的行引用哪个季度，一个 value 列是整个季度的值（以及每个日期范围内的季度），以及指示日期范围的 start_date 和 end_date 列。这是一个示例：

ID  quarter     value   start_date  end_date 
1   2            152    2019-11-07  2050-12-30
1   1            785    2019-11-07  2050-12-30
2   2            152    2019-03-05  2050-12-30
2   1            785    2019-03-05  2050-12-30
3   4            41     2018-06-12  2050-12-30
3   3            50     2018-06-12  2050-12-30
3   2            88     2018-06-12  2050-12-30
3   1            29     2018-06-12  2050-12-30
4   2           1607    2018-12-17  2050-12-30
4   1           4803    2018-12-17  2050-12-30

这是我想要的输出（ID 1）：

ID  quarter     value   start_date  end_date 
1   2            152/3  2020-04-01  2020-07-01
1   1            785/3  2020-01-01  2020-04-01
1   2            152/3  2021-04-01  2021-07-01
1   1            785/3  2021-01-01  2021-04-01

start_date 在输出中将是第一个表的下一个季度。我需要从第一个表的start_date 到end_date 生成系列。

Answer 1

这是解决问题的一种方法。根据您概述的输出显示示例。然后，您可以向 CASE/WHEN 添加更多条件以获取更多季度。

SELECT
    ID,
    Quarter,
    Value/3 AS "Value",
    CASE
     WHEN Quarter = 1 THEN '2020-01-01'
     WHEN Quarter = 2 THEN '2020-04-01'
    END AS "Start_Date",
    CASE
     WHEN Quarter = 1 THEN '2020-04-01'
     WHEN Quarter = 2 THEN '2020-07-01'
    END AS "End_Date"
   FROM
    Table

Answer 2

您可以通过使用GENERATE_SERIES 函数并传入每个唯一（ID）行的开始和结束日期并将间隔设置为 3 个月来完成此操作。然后将结果与ID 和季度上的原始表连接起来。

这是一个例子（注意original_data 是我称之为你的第一个表）：

WITH
quarters_table AS (
    SELECT
        t.ID,
        (EXTRACT('month' FROM t.quarter_date) - 1)::INT / 3 + 1 AS quarter,
        t.quarter_date::DATE AS start_date,
        COALESCE(
            LEAD(t.quarter_date) OVER (),
            DATE_TRUNC('quarter', t.original_end_date) + INTERVAL '3 months'
        )::DATE AS end_date
    FROM (
        SELECT
            original_record.ID,
            original_record.end_date AS original_end_date,
            GENERATE_SERIES(
                DATE_TRUNC('quarter', original_record.start_date),
                DATE_TRUNC('quarter', original_record.end_date),
                INTERVAL '3 months'
            ) AS quarter_date
        FROM (
            SELECT DISTINCT ON (original_data.ID)
                original_data.ID,
                original_data.start_date,
                original_data.end_date
            FROM
                original_data
            ORDER BY
                original_data.ID
        ) AS original_record
    ) AS t
)

SELECT
    quarters_table.ID,
    quarters_table.quarter,
    original_data.value::DOUBLE PRECISION / 3 AS value,
    quarters_table.start_date,
    quarters_table.end_date
FROM
    quarters_table
INNER JOIN
    original_data
ON
    quarters_table.ID = original_data.ID
    AND quarters_table.quarter = original_data.quarter;

样本输出：

 id | quarter |      value       | start_date |  end_date  
----+---------+------------------+------------+------------
  1 |       1 | 261.666666666667 | 2020-01-01 | 2020-04-01
  1 |       2 | 50.6666666666667 | 2020-04-01 | 2020-07-01
  1 |       1 | 261.666666666667 | 2021-01-01 | 2021-04-01
  1 |       2 | 50.6666666666667 | 2021-04-01 | 2021-07-01

为了完整起见，这是我在测试中使用的original_data 表：

WITH
original_data AS (
    SELECT
        1 AS ID,
        2 AS quarter,
        152 AS value,
        '2019-11-07'::DATE AS start_date,
        '2050-12-30'::DATE AS end_date
    
    UNION ALL

    SELECT
        1 AS ID,
        1 AS quarter,
        785 AS value,
        '2019-11-07'::DATE AS start_date,
        '2050-12-30'::DATE AS end_date
    
    UNION ALL

    SELECT
        2 AS ID,
        2 AS quarter,
        152 AS value,
        '2019-03-05'::DATE AS start_date,
        '2050-12-30'::DATE AS end_date
    
    -- ...
)