Question

在Amazon Athena中，我有一个像这样的表：

id   amount date
1    100    2018-04-05
1    50     2018-06-18
2    10     2018-04-23
2    100    2018-04-28
2    50     2018-07-07
2    10     2018-08-08

我想要一个类似

的结果

id   cum_sum date
1    100    2018-04
1    100    2018-05
1    150    2018-06
1    150    2018-07
1    150    2018-08
2    110    2018-04
2    110    2018-05
2    110    2018-06 
2    160    2018-07
2    170    2018-08

因此，我想获得每个月底（每个月的最后一天）每个ID的累计金额。我知道如何逐月执行此操作，但不会在一个查询中进行。

另一个问题也变成填写空的月份（即ID 1没有所有月份的条目，因此累计总和只能重用）。

如果还有MySQL的解决方案，我也将不胜感激。

我希望这是有道理的，在此先感谢。

Answer 1

您可以在PrestoDB中使用窗口功能。您可以生成日期。列出这些也很简单：

with months as (
      selecct '2018-04-01' as yyyy_mm union all    -- use the first of the month
      select '2018-05-01' union all
      select '2018-06-01' union all
      select '2018-07-01' union all
      select '2018-08-01'
     )
select i.id, m.yyyy_mm, sum(t.amt) as month_amount,
       sum(sum(t.amt)) over (partition by i.id order by m.yyyy_mm) as cumulative_amt
from (select distinct id from t) i cross join
     months m left join
     t
     on t.id = i.id and
        t.date >= m.yyyy_mm and
        t.date < m.yyyy_mm + interval '1 day'
group by i.id, m.yyyy_mm
order by i.id, m.yyyy_mm;

这在MySQL 8.0中也应该起作用。在早期版本中，您将需要变量或相关的子查询。第一个在PrestoDB中不起作用。第二个可能性能更差。

Answer 2

这是一个MySQL 8+解决方案，但是可以轻松地适应早期版本或支持CTE的另一个数据库。它使用日历表获取id的值和日期。跨月/ id生成金额后，然后进行累加总和以获得最终结果。

WITH ids AS (
    SELECT 1 AS id FROM dual UNION ALL
    SELECT 2 FROM dual
),
months AS (
    SELECT '2018-04-01' AS month UNION ALL    -- use the first of the month
    SELECT '2018-05-01' UNION ALL             -- to represent a given month
    SELECT '2018-06-01' UNION ALL
    SELECT '2018-07-01' UNION ALL
    SELECT '2018-08-01'
),
cte AS (
    SELECT
        i.id,
        m.month,
        SUM(amount) AS amount
    FROM ids i
    CROSS JOIN months m
    LEFT JOIN yourTable t
        ON t.id = i.id AND
           t.date >= m.month AND t.date < DATE_ADD(m.month, INTERVAL 1 MONTH)
    GROUP BY
        i.id,
        m.month
)

SELECT
    id,
    (SELECT SUM(t2.amount) FROM cte t2
     WHERE t1.id = t2.id AND t2.month <= t1.month) cum_sum,
    DATE_FORMAT(month, '%Y-%m') AS date
FROM cte t1
ORDER BY
    id,
    month;

Demo

要使以上代码在MySQL的早期版本或PrestoDB上运行，最大的挑战在于能否删除CTE，以及删除日期函数逻辑。除此之外，查询应保持不变。

Presto中按ID和按月的累计总和

2 个答案:

Demo