Spark sql 每组前 n 个

Question

如何在spark-sql 中获得每个组的前 n 名（比如说前 10 名或前 3 名）？

http://www.xaprb.com/blog/2006/12/07/how-to-select-the-firstleastmax-row-per-group-in-sql/ 提供通用 SQL 教程。但是，spark 并没有在 where 子句中实现子查询。

Answer 1

您可以使用 Spark 1.4 中添加的窗口函数功能 假设我们有一个如下所示的 productRevenue 表。

每个品类最畅销和第二畅销的产品是什么的答案如下

SELECT product,category,revenue FROM 
   (SELECT product,category,revenue,dense_rank() 
         OVER (PARTITION BY category ORDER BY revenue DESC) as rank 
    FROM productRevenue) tmp 
WHERE rank <= 2

这会给你想要的结果