讲师中心 微信公众号
AI工具推荐 视频效率加速

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

秋伟君_5077

秋伟君_5077

发布时间:2026-10-12 14:30:59

|

524人浏览过

|

来源于php中文网

原创

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<category:string,score:double>>。这种写法比拼 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?还是只需要计数、最大值、最近三个?多数时候,原始需求里藏着过度聚合的假设。

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

热门AI工具

更多
咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

讯飞绘文

讯飞绘文是一款由科大讯飞推出的一站式 AIGC 内容运营平台。

相关专题

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

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

4303

2023.10.12

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

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

911

2023.10.27

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

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

1109

2024.02.23

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

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

6201

2024.03.06

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

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

3043

2024.03.06

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

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

6180

2024.04.07

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

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

8361

2024.04.29

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

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

1130

2024.04.29

OpenJDK JVM参数怎么调优
OpenJDK JVM参数怎么调优

Java应用运行缓慢,如何调整JVM参数?本专题介绍OpenJDK JVM性能调优的基本流程,涵盖性能指标确认、运行数据采集、内存配置和垃圾回收分析,帮助你先定位瓶颈,再通过对照测试验证效果,避免直接套用参数而影响应用稳定性。

0

2026.10.16

热门下载

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

精品课程

更多
热门推荐
/
最新课程
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn