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

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

轻强大大_9276

轻强大大_9276

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

|

575人浏览过

|

来源于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:

python学习
python学习

Python 编程学习助手 skill,面向零基础用户。 当用户表达以下意图时触发: - 学习 Python 基础概念(变量、数据类型、循环、函数、类等) - 请求 Python 练习题或测验 - 请求审查或调试 Python 代码 - 请求 Python 项目实战指导(爬虫、数据分析、Web 开发等入门项目) - 泛化表述如"教我 Python""我想学编程""Python 怎么写""帮我学 Python" 此 skill

下载
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工具

更多
豆包大模型

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

PixPix
PixPix Hot

PixPix是一款面向电商视觉生产的AI商品图生成工具。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

DeepSeek

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

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

讯飞智作

讯飞智作是一款AI视频创作工具,AI文本配音工具,数字人课程、营销视频制作。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

WorkBuddy

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

切问学术

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

相关专题

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

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

4378

2023.08.10

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

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

1229

2025.12.24

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

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

423

2026.01.21

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

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

559

2026.01.21

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

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

409

2026.02.06

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

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

253

2026.03.16

Java 多线程与并发编程实战
Java 多线程与并发编程实战

深入讲解 Java 并发编程体系,涵盖 Thread / Runnable / Callable 线程创建方式、线程生命周期与状态转换、synchronized 同步锁与 Lock/ReentrantLock 显式锁、volatile 可见性保证、ThreadPool 线程池配置与调优(核心参数/拒绝策略)、JUC 并发工具类(CountDownLatch/CyclicBarrier/Semaphore/ConcurrentHashMa

369

2026.04.17

C++ 多线程使用教程大全
C++ 多线程使用教程大全

系统讲解 C++11 及后续标准引入的多线程编程能力,涵盖 std::thread 线程创建与生命周期管理、std::mutex / shared_mutex 互斥锁与读写锁、std::condition_variable 条件变量同步、std::atomic 原子操作与内存序(memory_order)、std::future / promise / async 异步任务、线程池的设计与实现、死锁检测与避免策略,帮助开发者编写正确高效

216

2026.04.28

Gradle插件开发入门教程
Gradle插件开发入门教程

本专题介绍Gradle插件开发的基本流程,涵盖插件类编写、自定义任务注册、扩展参数设计和插件应用方法,帮助你理解插件如何组织构建逻辑,减少多个项目之间的重复配置。

60

2026.10.22

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Uvicorn官方安装与入门手册
Uvicorn官方安装与入门手册

共0课时 | 0人学习

Python uv官方项目管理指南
Python uv官方项目管理指南

共0课时 | 0人学习

Python uv官方安装与入门文档
Python uv官方安装与入门文档

共0课时 | 0人学习

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

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