来源:字节今日头条
目录
- [1 题目](#1 题目)
- [2 建表语句](#2 建表语句)
- [3 题解](#3 题解)
1 题目
有如下数据,记录每天每只股票的收盘价格,请查出每只股票的波峰和波谷的日期和价格;
波峰定义:股票价格高于前一天和后一天价格时为波峰
波谷定义:股票价格低于前一天和后一天价格是为波谷
2 建表语句
sql
CREATE TABLE t_stock_test (
ts_code VARCHAR(20) COMMENT '股票代码',
trade_date VARCHAR(20) COMMENT '交易日期',
close FLOAT COMMENT '收盘价'
);
INSERT INTO `t_stock_test` VALUES
('000001.SZ','20220104',16.66),
('000002.SZ','20220104',20.49),
('000001.SZ','20220105',17.15),
('000002.SZ','20220105',21.17),
('000001.SZ','20220106',17.12),
('000002.SZ','20220106',21.05),
('000001.SZ','20220107',17.2),
('000002.SZ','20220107',21.89),
('000001.SZ','20220110',17.19),
('000002.SZ','20220110',22.16),
('000001.SZ','20220111',17.41),
('000002.SZ','20220111',22.3),
('000001.SZ','20220112',17),
('000002.SZ','20220112',22.05),
('000001.SZ','20220113',16.98),
('000002.SZ','20220113',21.53),
('000001.SZ','20220114',16.33),
('000002.SZ','20220114',20.7),
('000001.SZ','20220117',16.22),
('000002.SZ','20220117',20.87);
样例结果
3 题解
需要比较当天价格与前一天、后一天的价格进行比较,常规想法为进行关联,股票ID相等、日期为当天日期减1,为前一天价格,日期为当天价格加1,为后一天价格,然后进行计算;简化方法为使用 lag 和 lead 函数,可以避免进行表关联;
- lag()函数
- LAG(col,n,DEFAULT) 用于统计窗口内往上第 n 行。参数1为列名,参数2为往上第 n 行(可选,默认为1),参数3为默认值(当往上第n行为NULL时候,取默认值,如不指定,则为NULL)
- lead()函数
- LEAD(col,n,DEFAULT) 用于统计窗口内往下第 n 行。参数1为列名,参数2为往下第 n 行(可选,默认为1),参数3为默认值(当往下第n行为NULL时候,取默认值,如不指定,则为NULL)
sql
select
ts_code,
trade_date,
case when close> last_close and close>next_close then '波峰' else '波谷' end as type
from
(select
*,
lag(close,1)over(partition by ts_code order by trade_date) last_close,
lead(close,1)over(partition by ts_code order by trade_date) next_close
from t_stock_test)t
where (close> last_close and close>next_close)
or (close< last_close and close<next_close);