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

Polars 中使用 join_asof 实现分段映射与差值计算

胖枫小哥_6423

胖枫小哥_6423

发布时间:2026-01-04 22:20:21

|

741人浏览过

|

来源于php中文网

原创

Polars 中使用 join_asof 实现分段映射与差值计算

本文介绍如何在 polars 中高效生成新列 `baz`:对 dataframe 的 `bar` 列中每个值,从预排序列表 `points` 中查找**不超过该值的最大元素**,再计算二者之差。核心方案是利用 `join_asof` 的 `backward` 策略,全程基于表达式链式操作,无需 python 循环或 `apply`。

在 Polars 中处理此类“查找最近下界(floor lookup)+ 差值”任务时,最高效、最符合声明式风格的方式是使用 join_asof —— 它专为有序键的近似连接设计,性能远超逐行 apply 或 map_dict。

✅ 解决方案:join_asof + backward 策略

假设原始 DataFrame 为 df,且 points = [0, 1500, 3000, 4500, 6000, 7500, 9000, 10500, 12000](已严格升序),执行以下链式操作:

Li Python Sec Check
Li Python Sec Check

Python 安全规范检查工具:基于 CloudBase 规范、腾讯安全指南,LLM 智能分析(默认禁用,优先本地执行)

下载
points = [0, 1500, 3000, 4500, 6000, 7500, 9000, 10500, 12000]
df_points = pl.DataFrame({"point": points}).set_sorted("point")

result = (
    df.sort("bar")  # 关键:left_on 列需升序(join_asof 要求)
    .join_asof(df_points, left_on="bar", right_on="point")  # 默认 strategy="backward"
    .with_columns(baz=pl.col("bar") - pl.col("point"))
    .drop("point")
    .sort("foo")  # 恢复原始行序(可选)
)
? 原理说明: join_asof(..., strategy="backward") 对左表每行 bar=y,在右表 point 中查找满足 point ≤ y 的最大 point 值(即 floor)。 因 df_points 已通过 .set_sorted("point") 标记为有序,Polars 可跳过排序开销,直接二分查找,时间复杂度为 O(n log m)。 最终 baz = bar - point 即为所求差值。

⚠️ 注意事项

  • 必须保证 points 已升序,并调用 .set_sorted("point") 显式声明;否则 join_asof 行为未定义或报错。
  • join_asof 要求左表 left_on 列(此处为 "bar")在连接前已排序(.sort("bar") 不可省略),否则结果不正确。
  • 若 bar 中存在小于 points[0](即 0)的值,point 将为 null,导致 baz 也为 null。如需兜底(例如设为 bar 自身),可改用:
    .with_columns(baz=pl.col("bar") - pl.col("point").fill_null(0))

✅ 输出验证

结果与预期一致(部分截取):

┌─────┬───────┬──────┐
│ foo ┆ bar   ┆ baz  │
│ --- ┆ ---   ┆ ---  │
│ i64 ┆ i64   ┆ i64  │
╞═════╪═══════╪══════╡
│ 86  ┆ 11592 ┆ 1092 │  ← 11592 - 10500 = 1092
│ 109 ┆ 2765  ┆ 1265 │  ← 2765 - 1500 = 1265
│ 160 ┆ 1134  ┆ 1134 │  ← 1134 - 0 = 1134
└─────┴───────┴──────┘

此方法兼具高性能、可读性与可扩展性,是 Polars 生态中处理分段基准映射类问题的标准范式。

热门AI工具

更多
超级简历WonderCV

一款AI办公效率工具,主要用于免费求职简历模版下载制作,应届生职场人必备简历制作神器,适合需要提升相关任务效率的用户。

DeepSeek

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

立刻MV
立刻MV Hot

立刻MV是一款AI文本写作工具,AI 音乐视频(MV)创作工具。

豆包大模型

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

UpDream
UpDream Hot

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

WorkBuddy

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

LibLibAI
LibLibAI Hot

一款AI视频创作工具,主要用于国内领先的AI创意平台,以海量模型、低门槛操作与“创作-分享-商业化”生态,让小白与专业创作者都能高效实现图文乃至视频创意表达,适合需要提升相关任务效率的用户。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

VibeKnow
VibeKnow Hot

一款AI视频创作工具,主要用于全球首个AI知识视频创作平台,文档、文章、网页,一键生成视频,适合需要提升相关任务效率的用户。

相关专题

更多
c语言中null和NULL的区别
c语言中null和NULL的区别

c语言中null和NULL的区别是:null是C语言中的一个宏定义,通常用来表示一个空指针,可以用于初始化指针变量,或者在条件语句中判断指针是否为空;NULL是C语言中的一个预定义常量,通常用来表示一个空值,用于表示一个空的指针、空的指针数组或者空的结构体指针。

529

2023.09.22

java中null的用法
java中null的用法

在Java中,null表示一个引用类型的变量不指向任何对象。可以将null赋值给任何引用类型的变量,包括类、接口、数组、字符串等。想了解更多null的相关内容,可以阅读本专题下面的文章。

1678

2024.03.01

sort排序函数用法
sort排序函数用法

sort排序函数的用法:1、对列表进行排序,默认情况下,sort函数按升序排序,因此最终输出的结果是按从小到大的顺序排列的;2、对元组进行排序,默认情况下,sort函数按元素的大小进行排序,因此最终输出的结果是按从小到大的顺序排列的;3、对字典进行排序,由于字典是无序的,因此排序后的结果仍然是原来的字典,使用一个lambda表达式作为key参数的值,用于指定排序的依据。

1118

2023.09.04

LLVM自定义Pass怎么写
LLVM自定义Pass怎么写

本专题聚焦LLVM自定义Pass开发,整理Pass类结构、run()方法、PreservedAnalyses、CMake构建、插件注册、-load-pass-plugin加载和测试用例编写流程。

80

2026.09.30

LLVM RISC-V参数配置教程
LLVM RISC-V参数配置教程

本专题介绍LLVM对RISC-V基础ISA和扩展的支持方式,涵盖RV32、RV64、标准扩展、实验性扩展、厂商扩展、-menable-experimental-extensions和版本差异。

80

2026.09.30

LLVM IR中间表示入门指南
LLVM IR中间表示入门指南

本专题整理LLVM IR的核心概念,包括中间表示作用、模块结构、函数、基本块、SSA形式、类型系统和常见语法,帮助新手理解LLVM编译流程中的关键层。

80

2026.09.30

PDF转图片方法
PDF转图片方法

需要把 PDF 页面用于上传、预览、分享或图片归档时,PDF 转图片方法专题整理 JPG/PNG 格式选择、逐页导出、清晰度设置、批量下载和结果检查等流程,帮助用户稳定完成 PDF 图片化处理。

40

2026.09.30

PixTV AI视频生成与无限画布创作
PixTV AI视频生成与无限画布创作

PixTV专题整理AI视频与视觉内容创作相关功能使用教程,涵盖AI生图、视频生成、无限画布、多模型创作、素材管理、声音音乐及视频剪辑等功能,帮助用户快速掌握PixTV从创意到成片的完整制作方法。

60

2026.09.29

Buffalo框架数据库开发全教程
Buffalo框架数据库开发全教程

本专题围绕Buffalo框架数据库开发,讲解database.yml多环境配置、soda与fizz迁移生成回滚、模型结构体标签、增删改查与条件查询、一对多与多对多关联、数据校验、回调钩子、事务处理及原生SQL执行能力。

280

2026.09.23

热门下载

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

精品课程

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

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