时间戳

首页 > 解决方案 > SQL 计算基于 Hive 列中先前值重置的累积总和

sql - SQL 计算基于 Hive 列中先前值重置的累积总和

问题描述

我正在尝试用这样的东西创建一个累积值

KEY1    Date_    VAL1    CUMU_VAL2
K1      D1      1       0
K1      D2      1       1
K1      D3      0       2
K1      D4      1       0
K1      D5      1       1

因此，问题基本上是在 VAL1 中的前一行的基础上继续在 CUMU_VAL2 列中将值加 1，但是当 VAL1 列中的前一个值为零时，此总和会重置。基本上，如果你在 excel 中这样做，那么 Cell(D3) 的公式是

  D3 = IF(C2>0, D2+1, 0)

我相信我应该能够做到这样，但是当先前的值为零时如何添加案例然后重置总和？

SELECT
   a1.*,
       SUM(a1.VAL1) OVER (PARTITION BY a1.KEY1 ORDER  BY a1.Date_ ) AS CUMU_VAL2
FROM source_table a1

标签： sqlhivepartition-by

解决方案

您可以分配一个组 - 这是给定行之后的 0 的总和。然后使用count()：

select t.KEY1, t.Date_, t.VAL1,
       count(*) over (partition by key1, grp, (case when val1 = 0 then 0 else 1 end)
                      order by date_
                     ) as cume_val1
from (select t.*,
             sum(case when a.val1 = 0 then 1 else 0 end) over (partition by key1 order by date_ rows between 1 following and unbounded following) as grp
      from source_table t
     ) t;

如果val1只取值 0 和 1，则使用row_number()而不是count().

推荐阅读