本文介绍一种简洁、可扩展的 pyspark 聚合方案,通过 `collect_list + struct` 一次性捕获完整历史,再用 `filter` 和 `transform` 精准提取最新字段与结构化列表,避免多次窗口计算,显著提升多字段(如姓名、地址等)场景下的代码复用性与执行效率。
在实际数据处理中,常需对用户级(如 id)数据按时间戳聚合:既要保留全部历史快照(如每次更新的姓名、地址),又要快速获取最新状态(如最新姓名、最新地址、最新时间戳)。原始方案使用窗口函数 Window.partitionBy(“id”).orderBy(“timestamp”.desc()) 配合多次 first() 计算,虽可行,但在扩展至多个字段(如 address1, address2, address3)时,会导致重复定义窗口、冗余列计算和难以维护的链式 withColumn。
更优解是“一次收集、二次解析”范式:先用 groupBy + collect_list(struct(…)) 将每组所有行打包为结构化数组,再基于该数组做逻辑提取——既避免窗口开销,又天然支持任意字段组合。
以下为推荐实现(已适配您提供的示例数据):