如何在Spark SQL中用COLLECT_LIST实现集合聚合

秋伟君_5077

秋伟君_5077

2026-10-12

504人浏览

原创

collect_list保留重复元素和插入顺序,不自动去重或排序,null值正常收集,支持复杂类型,但大数据量易oom,需谨慎评估是否需全量聚合。

如何在spark sql中用collect_list实现集合聚合

COLLECT_LIST 会保留重复元素和插入顺序

Spark SQL 的 COLLECT_LIST 不是去重操作,它按数据实际到达的分区和行序把值攒成数组,相同值多次出现就会多次出现在结果里。如果你需要去重,得额外套一层 ARRAY_DISTINCT 或改用 COLLECT_SET ——但后者不保证顺序。

常见错误是误以为 COLLECT_LIST 等价于 Python 的 set() 或 SQL 的 DISTINCT,结果聚合后发现数组里一堆重复项,还纳闷“为什么没自动 dedup”。

  • 它在 shuffle 阶段按分组 key 拉取所有匹配行,不做任何过滤或排序预处理
  • 若上游用了 ORDER BY(比如在窗口函数中),顺序可被控制;但单纯 GROUP BY + COLLECT_LIST 不保证全局顺序
  • 空值(NULL)会被正常收集进数组,不会跳过

在 GROUP BY 场景下直接使用 COLLECT_LIST

这是最常见用法:对每组生成一个元素列表。注意列名必须明确,不能写 *,且非聚合列必须出现在 GROUP BY 子句中。

SELECT user_id, COLLECT_LIST(item_id) AS items
FROM events
GROUP BY user_id

如果想让列表按时间排序,得先排序再聚合:

SELECT user_id, COLLECT_LIST(item_id) AS items
FROM (
  SELECT user_id, item_id
  FROM events
  ORDER BY event_time
) t
GROUP BY user_id
  • 子查询里的 ORDER BY 在 Spark 中仅对单个 task 有效,跨 partition 仍不保序;更稳妥的做法是用窗口函数加 ROW_NUMBER 标序号,再在外层按序号排序后聚合
  • 若 item_id 是字符串类型,结果数组元素类型自动推导为 array<string></string>;数值列同理,类型一致
  • 遇到内存压力大时(比如某 user_id 关联百万级 item_id),COLLECT_LIST 可能触发 OOM,此时应考虑采样或限制长度

用 COLLECT_LIST 处理嵌套结构或复杂类型

它支持结构体(struct)、map、甚至其他 array 类型。例如把多行属性打包成结构体列表:

SELECT user_id,
       COLLECT_LIST(NAMED_STRUCT('category', category, 'score', score)) AS preferences
FROM user_scores
GROUP BY user_id

结果是 array<struct>></struct>。这种写法比拼 JSON 字符串更安全,下游可用 TRANSFORM 或字段访问符(如 preferences.category)进一步展开。

  • 结构体内字段名必须合法(不能含空格或特殊符号),否则报错 Invalid column reference
  • 如果某行中 category 为 NULL,整个 struct 仍被加入数组,只是对应字段为 NULL
  • 嵌套太深(比如 array of struct of array)会让序列化开销明显上升,执行计划里能看到 SerializeFromObject 耗时突增

性能与替代方案:什么时候不该用 COLLECT_LIST

当分组后单条输出的数组长度超过几万,或者总数据量超亿级时,COLLECT_LIST 很容易成为瓶颈。它的本质是把分布式数据全拉到一个 task 做合并,反向放大 shuffle 压力。

  • 替代思路一:用 STRING_AGG 拼接成字符串(更省内存,但失去类型信息)
  • 替代思路二:用 AGGREGATE + 自定义初始值和合并逻辑,控制中间状态大小(比如只保留 top-N)
  • 替代思路三:改用广播 join + 预聚合表,把集合构建提前到 ETL 阶段,查询时直接读物化结果

真正难的不是写对语法,而是判断“这个列表到底要不要全量收集”——业务上是否真的需要全部 item_id?还是只需要计数、最大值、最近三个?多数时候,原始需求里藏着过度聚合的假设。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

2023.10.12

4183

8

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.27

891

4

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

2024.02.23

1089

5

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

6081

10

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

2943

4

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

2024.04.07

6040

11

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

2024.04.29

8121

6

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1110

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

972

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习