0

0

Polars 中按连续行分组聚合的惯用方法

花韻仙語

花韻仙語

发布时间:2026-01-30 15:43:12

|

535人浏览过

|

来源于php中文网

原创

Polars 中按连续行分组聚合的惯用方法

在 polars 中,对 dataframe 按固定长度(如每 n 行)进行滑动或分段聚合时,`group_by_dynamic` 配合行索引是最推荐、最高效且符合惯用法的方案,优于手动构造分组键或使用窗口函数。

当需要将大型 DataFrame(例如 10⁶ 行、多列)按连续 n 行(如 n = 100)分段并执行聚合(如均值、求和、标准差等)时,核心挑战在于:如何简洁、高效、无副作用地构造分组依据。Polars 官方明确推荐 group_by_dynamic 用于此类“基于索引的动态分组”,它专为时间序列或有序整数索引设计,底层高度优化,支持并行分组计算,避免了 Python 层循环或 apply 的性能损耗。

你当前使用的 pl.int_range(pl.len()) 方式完全正确,但可进一步简化与健壮化——推荐使用 .with_row_index(),它更语义清晰、自动处理空表边界,并且是 Polars 0.20.0+ 的标准 API:

Ai好记
Ai好记

强大的AI音视频转录与总结工具

下载
import polars as pl

df = pl.DataFrame({"a": [1, 1, 3, 8, 62, 535, 4213], "b": range(7)})

n = 3
result = (
    df
    .with_row_index()
    .group_by_dynamic(
        index_column=pl.col("index").cast(pl.Int32),  # 必须为有符号整数
        every=f"{n}i",                                 # "3i" 表示每 3 行一组
        period=f"{n}i",                                 # 可选:若需重叠(如滑动窗口),设为 period="3i", offset="1i"
        closed="left"                                   # 默认行为:[0,3), [3,6), [6,9)...
    )
    .agg(
        pl.col("a").mean().alias("a_mean"),
        pl.col("b").sum().alias("b_sum"),
        pl.col("a").std().alias("a_std")
    )
)
print(result)
✅ 关键要点说明: every="ni" 中的 "i" 后缀表示 integer-based 分组(非时间),这是处理行号分组的必需写法; index_column 必须是 Int32 或 Int64(有符号),因此 .cast(pl.Int32) 不可省略(.with_row_index() 默认返回 UInt32); 若需滑动窗口(如每 1 行移动、取连续 100 行),请改用 rolling + over(如 pl.col("a").rolling_mean(window_size=n).over(pl.int_range(0, pl.len()) // n)),但注意其内存开销更大;group_by_dynamic 始终适用于不重叠分块聚合,性能最优; 对于超大表(10⁶+ 行),该模式天然支持 Polars 的惰性执行与多线程调度,无需额外优化。

综上,你的原始思路完全符合 Polars 惯用法——只需将索引构造升级为 .with_row_index() 并显式类型转换,即可获得更简洁、可读性更强且生产就绪的实现。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
线程和进程的区别
线程和进程的区别

线程和进程的区别:线程是进程的一部分,用于实现并发和并行操作,而线程共享进程的资源,通信更方便快捷,切换开销较小。本专题为大家提供线程和进程区别相关的各种文章、以及下载和课程。

806

2023.08.10

Python 多线程与异步编程实战
Python 多线程与异步编程实战

本专题系统讲解 Python 多线程与异步编程的核心概念与实战技巧,包括 threading 模块基础、线程同步机制、GIL 原理、asyncio 异步任务管理、协程与事件循环、任务调度与异常处理。通过实战示例,帮助学习者掌握 如何构建高性能、多任务并发的 Python 应用。

381

2025.12.24

java多线程相关教程合集
java多线程相关教程合集

本专题整合了java多线程相关教程,阅读专题下面的文章了解更多详细内容。

33

2026.01.21

C++多线程相关合集
C++多线程相关合集

本专题整合了C++多线程相关教程,阅读专题下面的的文章了解更多详细内容。

31

2026.01.21

C# 多线程与异步编程
C# 多线程与异步编程

本专题深入讲解 C# 中多线程与异步编程的核心概念与实战技巧,包括线程池管理、Task 类的使用、async/await 异步编程模式、并发控制与线程同步、死锁与竞态条件的解决方案。通过实际项目,帮助开发者掌握 如何在 C# 中构建高并发、低延迟的异步系统,提升应用性能和响应速度。

107

2026.02.06

C++多线程并发控制与线程安全设计实践
C++多线程并发控制与线程安全设计实践

本专题围绕 C++ 在高性能系统开发中的并发控制技术展开,系统讲解多线程编程模型与线程安全设计方法。内容包括互斥锁、读写锁、条件变量、原子操作以及线程池实现机制,同时结合实际案例分析并发竞争、死锁避免与性能优化策略。通过实践讲解,帮助开发者掌握构建稳定高效并发系统的关键技术。

12

2026.03.16

Python 多线程与异步编程实战
Python 多线程与异步编程实战

本专题系统讲解 Python 多线程与异步编程的核心概念与实战技巧,包括 threading 模块基础、线程同步机制、GIL 原理、asyncio 异步任务管理、协程与事件循环、任务调度与异常处理。通过实战示例,帮助学习者掌握 如何构建高性能、多任务并发的 Python 应用。

381

2025.12.24

java多线程相关教程合集
java多线程相关教程合集

本专题整合了java多线程相关教程,阅读专题下面的文章了解更多详细内容。

33

2026.01.21

抖漫入口地址合集
抖漫入口地址合集

本专题整合了抖漫入口地址相关合集,阅读专题下面的文章了解更多详细地址。

17

2026.03.17

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 2万人学习

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

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