文档简介:
窗口函数
窗口函数是一类特殊的内置函数。和聚合函数类似,窗口函数也是对于多个输入行做计算得到一个数据值。不同的是,窗口函数是在一个特定的窗口内对输入数据做处理,而不是按照 group by 来分组计算。并且窗口函数的输入和输出行是一对一的关系,而不是多对一的关系。 每个窗口内的数据可以用 over() 从句进行排序和分组。窗口函数会对结果集的每一行计算出一个单独的值,而不是每个 group by 分组计算一个值。这种灵活的方式允许用户在 select 从句中增加额外的列,给用户提供了更多的机会来对结果集进行重新组织和过滤。
窗口函数只能出现在 select 列表以及最外层的 order by 从句中。在查询过程中,窗口函数会在最后生效,就是说,在执行完join,where 和 group by 等操作之后再执行。窗口函数在金融和科学计算领域经常被使用到,用来分析趋势、计算离群值以及对大量数据进行分桶分析等。
窗口函数语法
function(args) OVER(partition_by_clause order_by_clause [window_clause])
partition_by_clause:
PARTITION BY expr [, expr ...]
order_by_clause:
ORDER BY expr [ASC | DESC] [, expr [ASC | DESC] ...]
window_clause:
ROWS BETWEEN [ { m | UNBOUNDED } PRECEDING | CURRENT ROW] [ AND [CURRENT ROW | { UNBOUNDED | n } FOLLOWING] ]
RANGE BETWEEN [ {m | UNBOUNDED } PRECEDING | CURRENT ROW] [ AND [CURRENT ROW | { UNBOUNDED | n } FOLLOWING] ]
-
function
具体的窗口函数。目前支持的 Function 包括:
- AVG()
- COUNT()
- DENSE_RANK()
- FIRST_VALUE()
- LAG()
- LAST_VALUE()
- LEAD()
- MAX()
- MIN()
- RANK()
- ROW_NUMBER()
- SUM()
-
partition_by_clause
Partition By 从句和 Group By 类似,它把输入行按照指定的一列或多列分组,相同值的行会被分到一组。
每一组可以理解为一个窗口。如果不指定 Partition By 从句,则整个数据集为一组,即一个窗口。
-
order_by_clause
Order By 从句用于指定每一个分组内的数据排列顺序。数据集先按 Partition By 子句分组,每组内再排序,排序后按 Function 对每行进行计算。
与外层 Order By 的不同点是,OVER从句中的 Order By n(n是正整数) 相当于不做任何操作,而外层的 Order By n 表示按照第n列排序。
举例说明:
SELECT row_number() OVER (PARTITION BY class_id ORDER BY eventdate) AS id, c1, c2, c3, c4 FROM events;
这个SQL会将 events 表中的数据行按 class_id 划分成多组后,每组内的数据按 eventdate 列排序。对排序后的每组内的数据,添加行号 (1、2、3、4、...)。
-
window_clause
Window从句用来为窗口函数指定一个运算范围,以当前行为准,前后若干行作为窗口函数运算的对象。
Window从句支持的方法有:
- AVG()
- COUNT()
- FIRST_VALUE()
- LAST_VALUE()
- SUM()
- MAX()
- MIN()
对于 MAX() 和 MIN(), window从句可以指定开始范围 UNBOUNDED PRECEDING
假设我们有如下的股票数据,股票代码是JDR,closing price是每天的收盘价。
> create table stock_ticker (stock_symbol string, closing_price decimal(8,2), closing_date
-
timestamp); > ...load some data... > select * from stock_ticker order by stock_symbol, closing_date
-
| stock_symbol | closing_price | closing_date | |--------------|---------------|-----------------
-
----| | JDR | 12.86 | 2014-10-02 00:00:00 | | JDR | 12.89 | 2014-10-03 00:00:00 | | JDR | 12.94
-
| 2014-10-04 00:00:00 | | JDR | 12.55 | 2014-10-05 00:00:00 | | JDR | 14.03 | 2014-10-06 00:00:00
-
| | JDR | 14.75 | 2014-10-07 00:00:00 | | JDR | 13.98 | 2014-10-08 00:00:00 | ```
-
以下查询使用窗口函数产生 `moving_average` 这一列,它的值是3天的股票均价,即前一天、当前以及后一天三天的均价。 第一天没有前一天的值,最后一天没有后一天的值,所以这两行只计算了两天的均值。 这里 Partition By 没有起到作用,因为所有的数据都是JDR的数据,但如果还有其他股票信息,Partition By
-
会保证窗口函数值作用在本Partition之内。 ```sql > select stock_symbol, closing_date, closing_price,
-
avg(closing_price) over (partition by stock_symbol order by closing_date rows between 1 preceding
-
and 1 following) as moving_average from stock_ticker; | stock_symbol | closing_date | closing_price
-
| moving_average | |--------------|---------------------|---------------|----------------| |
-
JDR | 2014-10-02 00:00:00 | 12.86 | 12.87 | | JDR | 2014-10-03 00:00:00 | 12.89 | 12.89 | | JDR
-
| 2014-10-04 00:00:00 | 12.94 | 12.79 | | JDR | 2014-10-05 00:00:00 | 12.55 | 13.17 | | JDR
-
| 2014-10-06 00:00:00 | 14.03 | 13.77 | | JDR | 2014-10-07 00:00:00 | 14.75 | 14.25 | | JDR
-
| 2014-10-08 00:00:00 | 13.98 | 14.36 | ```
Function 介绍
本节介绍PALO中可以用作窗口函数的方法。
AVG()
计算平均值。
语法:
AVG([DISTINCT | ALL] *expression*) [OVER (*analytic_clause*)]
举例:
计算当前行和它前后各一行数据的x平均值
> select x, property, avg(x) over (partition by property order by x rows between 1 preceding and 1
following) as 'moving average' from int_t where property in ('odd','even'); | x | property | moving average
| |----|----------|----------------| | 2 | even | 3 | | 4 | even | 4 | | 6 | even | 6 | | 8 | even | 8
| | 10 | even | 9 | | 1 | odd | 2 | | 3 | odd | 3 | | 5 | odd | 5 | | 7 | odd | 7 | | 9 | odd | 8 |
该 SQL 首先会对数据集按照 property 分组,每组内按 x 列进行排序。之后对每组内的每行数据,计算其前一行,自身和后一行三行数据的平均值,作为该行的输出,写入 moving average 列。
COUNT()
统计个数。
语法:
COUNT([DISTINCT | ALL] expression) [OVER (analytic_clause)]
举例:
计算从当前行到第一行x出现的次数。
> select x, property,
count(x) over
(partition by property order by x rows between unbounded preceding and current row) as 'cumulative total'
from int_t where property in ('odd','even');
| x | property | cumulative count |
|----|----------|------------------|
| 2 | even | 1 |
| 4 | even | 2 |
| 6 | even | 3 |
| 8 | even | 4 |
| 10 | even | 5 |
| 1 | odd | 1 |
| 3 | odd | 2 |
| 5 | odd | 3 |
| 7 | odd | 4 |
| 9 | odd | 5 |
该 SQL 首先会对数据集按照 property 分组,每组内按 x 列进行排序。之后对每组内的每行数据,统计从自身算起之前所有行的行数,作为该行的输出,写入 cumulative count 列。
RANK()
RANK() 函数用来表示排名,与 DENSE_RANK()不同的是,如果出现了两个并列的值,RANK() 的第三个数就是 3,而不是 2。
语法:
RANK() OVER(partition_by_clause order_by_clause)
举例:
根据x列进行排名
> select x, y, rank() over(partition by x order by y) as rank from int_t; | x | y | rank | |---|---|--
----| | 1 | 1 | 1 | | 1 | 2 | 2 | | 1 | 2 | 2 | | 2 | 1 | 1 | | 2 | 2 | 2 | | 2 | 3 | 3 | |
3 | 1 | 1 | | 3 | 1 | 1 | | 3 | 2 | 3 |
该 SQL 首先会对数据集按照 x 分组,每组内按 y 列进行排序。之后对每组内的每行数据,计算该行在组内的名次,写入 rank 列。
DENSE_RANK()
DENSE_RANK() 函数用来表示排名,与 RANK() 不同的是,DENSE_RANK() 不会出现空缺数字。比如,如果出现了两个并列的1,DENSE_RANK() 的第三个数仍然是2,而 RANK() 的第三个数是3。
语法:
DENSE_RANK() OVER(partition_by_clause order_by_clause)
举例:
下例展示了按照property列分组对x列排名:
> select x, y, dense_rank() over(partition by x order by y) as rank from int_t; | x | y | rank
| |----|------|----------| | 1 | 1 | 1 | | 1 | 2 | 2 | | 1 | 2 | 2 | | 2 | 1 | 1 | | 2 | 2 | 2
| | 2 | 3 | 3 | | 3 | 1 | 1 | | 3 | 1 | 1 | | 3 | 2 | 2 |
该 SQL 首先会对数据集按照 x 分组,每组内按 y 列进行排序。之后对每组内的每行数据,计算该行在组内的名次,写入 rank 列。
FIRST_VALUE()
返回窗口范围内的第一个值。
语法:
FIRST_VALUE(expr) OVER(partition_by_clause order_by_clause [window_clause])
举例:
我们有如下数据
> select name, country, greeting from mail_merge;
| name | country | greeting |
|---------|---------|--------------|
| Pete | USA | Hello |
| John | USA | Hi |
| Boris | Germany | Guten tag |
| Michael | Germany | Guten morgen |
| Bjorn | Sweden | Hej |
| Mats | Sweden | Tja |
使用 FIRST_VALUE(),根据 country 分组,返回每个分组中第一个 greeting 的值:
> select country, name, first_value(greeting) over (partition by country order by name, greeting)
as greeting from mail_merge; | country | name | greeting | |---------|---------|-----------| | Germany
| Boris | Guten tag | | Germany | Michael | Guten tag | | Sweden | Bjorn | Hej | | Sweden | Mats | Hej
| | USA | John | Hi | | USA | Pete | Hi |
LAST_VALUE()
返回窗口范围内的最后一个值。与 FIRST_VALUE() 相反。
语法:
LAST_VALUE(expr) OVER(partition_by_clause order_by_clause [window_clause])
使用 FIRST_VALUE() 举例中的数据:
> select country, name, last_value(greeting) over (partition by country order by name, greeting)
as greeting from mail_merge; | country | name | greeting | |---------|---------|--------------|
| Germany | Boris | Guten morgen | | Germany | Michael | Guten morgen | | Sweden | Bjorn | Tja | |
Sweden | Mats | Tja | | USA | John | Hello | | USA | Pete | Hello |
LAG()
用来计算当前行向前数若干行的值。
语法:
LAG(expr, offset, default) OVER (partition_by_clause order_by_clause)
举例:
计算前一天的收盘价
> select stock_symbol, closing_date, closing_price, lag(closing_price, 1, 0) over (partition by
stock_symbol order by closing_date) as "yesterday closing" from stock_ticker order by closing_date;
| stock_symbol | closing_date | closing_price | yesterday closing | |--------------|-----------------
----|---------------|-------------------| | JDR | 2014-09-13 00:00:00 | 12.86 | 0 | | JDR |
2014-09-14 00:00:00 | 12.89 | 12.86 | | JDR | 2014-09-15 00:00:00 | 12.94 | 12.89 | | JDR |
2014-09-16 00:00:00 | 12.55 | 12.94 | | JDR | 2014-09-17 00:00:00 | 14.03 | 12.55 | | JDR |
2014-09-18 00:00:00 | 14.75 | 14.03 | | JDR | 2014-09-19 00:00:00 | 13.98 | 14.75 |
该 SQL 首先会对数据集按照 stock_symbol 分组,每组内按 closing_date 列进行排序。之后对每组内的每行数据,取其前一行的值作为输出,写入 yesterday closing 列。如果没有前一行,则使用默认值 0。
LEAD()
和 LAG() 方法相反。用来计算当前行向后数若干行的值。
语法:
LEAD (expr, offset, default]) OVER (partition_by_clause order_by_clause)
举例:
计算第二天的收盘价对比当天收盘价的走势,即第二天收盘价比当天高还是低。
> select stock_symbol, closing_date, closing_price, case (lead(closing_price, 1, 0) over
(partition by stock_symbol order by closing_date) - closing_price) > 0 when true then "higher"
when false then "flat or lower" end as "trending" from stock_ticker order by closing_date;
| stock_symbol | closing_date | closing_price | trending | |--------------|------------------
---|---------------|---------------| | JDR | 2014-09-13 00:00:00 | 12.86 | higher | | JDR |
2014-09-14 00:00:00 | 12.89 | higher | | JDR | 2014-09-15 00:00:00 | 12.94 | flat or lower
| | JDR | 2014-09-16 00:00:00 | 12.55 | higher | | JDR | 2014-09-17 00:00:00 | 14.03 | higher
| | JDR | 2014-09-18 00:00:00 | 14.75 | flat or lower | | JDR | 2014-09-19 00:00:00 | 13.98 | flat or lower |
该 SQL 首先会对数据集按照 stock_symbol 分组,每组内按 closing_date 列进行排序。之后对每组内的每行数据,取其后一行的值作为输出,再减去该行的 closing_price 值。如果没有后一行,则使用默认值 0。
MAX()
计算分组内的最大值。
语法:
MAX([DISTINCT | ALL] expression) [OVER (analytic_clause)]
举例:
计算从第一行到当前行之后一行的最大值
> select x, property, max(x) over (order by property, x rows between unbounded preceding and 1
following) as 'local maximum' from int_t where property in ('prime','square'); | x | property |
local maximum | |---|----------|---------------| | 2 | prime | 3 | | 3 | prime | 5 | | 5 | prime
| 7 | | 7 | prime | 7 | | 1 | square | 7 | | 4 | square | 9 | | 9 | square | 9 |
该 SQL 的窗口函数没有分组子句,则全部在一个组内。每组内按 property 和 x 列进行排序。之后对每组内的每行数据,取之前的所有行、自身和之后的1行,计算最大值作为输出写入 local maximum 列。
MIN()
计算分组内的最小值。
语法:
MIN([DISTINCT | ALL] expression) [OVER (analytic_clause)]
ROW_NUMBER()
为每个分组的每一行返回一个从1开始连续递增的整数。类似行号。
语法:
ROW_NUMBER() OVER(partition_by_clause order_by_clause)
举例:
> select x, y, row_number() over(partition by x order by y) as rank from int_t; | x | y | rank | |---|-
--|------| | 1 | 1 | 1 | | 1 | 2 | 2 | | 1 | 2 | 3 | | 2 | 1 | 1 | | 2 | 2 | 2 | | 2 | 3 | 3 | | 3 |
1 | 1 | | 3 | 1 | 2 | | 3 | 2 | 3 |
SUM()
分组内求和。
语法:
SUM([DISTINCT | ALL] expression) [OVER (analytic_clause)]
举例:
按照property进行分组,在组内计算当前行以及前后各一行的x列的和。
> select x, property, sum(x) over (partition by property order by x rows between 1 preceding and 1 following)
as 'moving total' from int_t where property in ('odd','even'); | x | property | moving total | |----|----
------|--------------| | 2 | even | 6 | | 4 | even | 12 | | 6 | even | 18 | | 8 | even | 24 | | 10 | even
| 18 | | 1 | odd | 4 | | 3 | odd | 9 | | 5 | odd | 15 | | 7 | odd | 21 | | 9 | odd | 16 |
该 SQL 首先会对数据集按照 property 分组,每组内按 x 列进行排序。之后对每组内的每行数据,取其前一行、自身和后一行的值求和,结果写入 moving total 列。