如何用单条SQL处理权重和为0的滑动窗口加权平均计算？

云原生阿岚

2026-6-12

解决方案

针对你的需求，我们可以利用DolphinDB的滑动窗口函数结合条件判断，用单条SQL实现符合要求的加权平均计算，处理权重和为0时返回NULL的场景。

核心思路

计算权重总和sum_w，用于后续判断分母是否为0；
用滑动窗口函数计算加权乘积之和（即公式中的分子）；
通过iif函数做条件分支：
- 若权重和为0，直接返回NULL；
- 若数据行不足以形成完整的4元素窗口（前3条数据），返回NULL；
- 其他情况返回加权平均值（分子/权重和）。

具体SQL实现

假设你的表名为t，权重向量为w（可根据每小时调度任务动态修改），示例代码如下：

-- 定义权重向量（可按需动态调整）
def w = [0.2, 0.3, 0.4, 0.1];
-- 计算权重总和
def sum_w = sum(w);

-- 执行查询
select 
    id,
    x,
    -- 条件判断：窗口不足或权重和为0时返回NULL，否则计算加权平均
    iif(id < 3 or sum_w == 0, NULL, movsum(x * reverse(w), 4) / sum_w) as y
from t
order by id;

代码解释

reverse(w)：调整权重顺序，使其与滑动窗口内的元素顺序匹配，确保加权和的计算符合公式y[i] = w[0]*x[i] + w[1]*x[i-1] + w[2]*x[i-2] + w[3]*x[i-3]；
movsum(x * reverse(w), 4)：计算滑动窗口内的加权乘积之和，即公式中的分子；
iif(...)：实现条件分支处理特殊场景，保证结果的正确性；
order by id：确保数据按id递增顺序处理，保证滑动窗口的逻辑正确。

拓展实现（自定义函数方式）

如果需要更直观的窗口计算逻辑，可以用自定义函数结合mov函数实现：

def calcWeightedSum(window, weights) {
    sum(window * reverse(weights))
}

def w = [0.2, 0.3, 0.4, 0.1];
def sum_w = sum(w);

select 
    id,
    x,
    iif(id < 3 or sum_w == 0, NULL, mov(x, 4, calcWeightedSum{w}) / sum_w) as y
from t
order by id;