0

0

如何在 Spark 中对分组数据集整体进行灵活处理

霞舞

霞舞

发布时间:2026-01-19 17:26:02

|

260人浏览过

|

来源于php中文网

原创

如何在 Spark 中对分组数据集整体进行灵活处理

本文介绍使用 spark 窗口函数(window + row_number)配合聚合操作,实现对每个分组内行序敏感的筛选与重组,例如跳过每组第二条记录后拼接剩余文本,无需自定义 udaf 即可高效完成。

在 Spark SQL 和 DataFrame API 中,若需对每个分组(如按 Names 分组)执行基于行序的精细操作(如“删除每组第 2 条记录”、“保留前 N 条”、“取中间值”等),直接使用 groupBy().agg() 无法满足需求——因为标准聚合函数(如 collect_list, concat_ws)不感知行内顺序。此时,窗口函数(Window Function)是更简洁、高效且符合 Spark 原生范式的解决方案

核心思路分为三步:

  1. 添加行号标识:使用 row_number().over(Window.partitionBy("Names").orderBy(...)) 为每组内各行分配唯一序号;
  2. 过滤目标行:通过 .where("rn 2") 排除每组中序号为 2 的记录;
  3. 重新聚合:对清洗后的数据按 Names 分组,用 collect_list 收集剩余文本,并以空格为分隔符拼接。

以下是完整 Java 实现(适配 Spark 3.x):

Sesame AI
Sesame AI

一款开创性的语音AI伴侣,具备先进的自然对话能力和独特个性。

下载
import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row;
import org.apache.spark.sql.SparkSession;
import org.apache.spark.sql.expressions.Window;
import static org.apache.spark.sql.functions.*;

// 假设 df 已初始化为原始 Dataset<Row>
Dataset<Row> result = df
    .withColumn("rn", 
        row_number().over(Window.partitionBy("Names").orderBy("Names"))) // 注意:orderBy 字段影响行序稳定性
    .filter(col("rn").notEqual(2))
    .groupBy("Names")
    .agg(concat_ws(" ", collect_list("Random_Text")).as("Random_Text"));
result.show();

关键说明与注意事项

  • orderBy("Names") 在窗口中仅用于保证 row_number 可重复性(因 Names 相同,实际顺序未定义)。若业务要求严格顺序(如按原始输入顺序或时间戳),应改用有区分度的列(如 orderBy("timestamp") 或添加 monotonically_increasing_id() 作为辅助排序键);
  • filter(col("rn").notEqual(2)) 等价于 .where("rn 2"),推荐使用列对象语法以增强类型安全;
  • concat_ws(" ", collect_list(...)) 自动忽略 null 值,适合文本拼接场景;若需更复杂逻辑(如去重、截断、格式化),可在 collect_list 后接 udf 或使用高阶函数(Spark 3.4+ 支持 transform, filter 等);
  • 无需自定义 UDAF:本方案完全基于内置函数,性能优、可读性强、易于维护,避免了 UDAF 开发、序列化及调试成本。

该模式可轻松扩展至其他分组级变换场景,例如:“每组保留最后 3 条”、“跳过首尾各 1 条”、“仅合并偶数行”等——只需调整窗口内的 orderBy 和后续 filter 条件即可。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

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

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

1133

2023.10.12

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

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

340

2023.10.27

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

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

381

2024.02.23

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

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

2152

2024.03.06

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

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

380

2024.03.06

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

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

1683

2024.04.07

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

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

585

2024.04.29

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

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

440

2024.04.29

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

3

2026.03.11

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 4.3万人学习

C# 教程
C# 教程

共94课时 | 11.1万人学习

Java 教程
Java 教程

共578课时 | 80.7万人学习

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号