0

0

Polars 中高效实现字符串部分匹配过滤的完整指南

花韻仙語

花韻仙語

发布时间:2026-03-18 11:39:19

|

441人浏览过

|

来源于php中文网

原创

Polars 中高效实现字符串部分匹配过滤的完整指南

本文介绍如何在 Polars 中高效过滤 DataFrame 行——基于另一 DataFrame 中多个子串进行批量、非循环的模糊匹配(如 SQL 的 LIKE '%pattern%'),避免逐行迭代,充分利用 Polars 的向量化能力和表达式优化。

本文介绍如何在 polars 中高效过滤 dataframe 行——基于另一 dataframe 中多个子串进行批量、非循环的模糊匹配(如 sql 的 `like '%pattern%'`),避免逐行迭代,充分利用 polars 的向量化能力和表达式优化。

在数据清洗与 ETL 场景中,常需从主表中剔除“文件名含关键词”的行(例如跳过所有以 skip 或 discard 开头的文件)。SQL 可通过 LIKE + 子查询高效完成,但若用传统 Pandas 或错误使用 Polars 循环(如 iter_rows()),不仅代码冗长,更会严重损害性能——失去 Polars 原生并行与查询优化优势。

幸运的是,Polars 自 0.20.0 起提供了专为批量子串匹配设计的 .str.contains_any() 方法,它能将目标列与一个字符串序列(如 Series)进行向量化模糊匹配,语义等价于「该字符串包含任意一个关键词」,天然适配此类过滤需求。

以下以实际数据为例演示标准做法:

假设你有两份 CSV 文件:

  • data.csv(主数据):

    filename,col2
    keep.txt,bar
    skip.txt,foo
    keep2.txt,zoom
    skip3.txt,custom1
    discard.txt,custom2
    file3.txt,custom3
    discard2.txt,custom4
    file4.txt,custom5
  • filter.csv(过滤关键词):

    skip
    discard

推荐方案:使用 .str.contains_any()(简洁、高效、原生支持)

import polars as pl

df_data = pl.read_csv("data.csv")
df_filter = pl.read_csv("filter.csv")

# 构建过滤表达式:保留 filename 不包含 filter 中任一关键词的行
result = df_data.filter(
    ~pl.col("filename").str.contains_any(df_filter["skip"])
)

print(result)

输出:

AIPURE
AIPURE

AIPURE帮您轻松找到2024年最佳AI工具

下载
shape: (4, 2)
┌───────────┬─────────┐
│ filename  ┆ col2    │
│ ---       ┆ ---     │
│ str       ┆ str     │
╞═══════════╪═════════╡
│ keep.txt  ┆ bar     │
│ keep2.txt ┆ zoom    │
│ file3.txt ┆ custom3 │
│ file4.txt ┆ custom5 │
└───────────┴─────────┘

? 原理说明
contains_any() 接收一个 Series[str](此处为 df_filter["skip"]),内部自动编译为高效正则或哈希查找逻辑(取决于引擎),对 filename 列单次全量扫描即可完成全部关键词匹配,时间复杂度接近 O(n),远优于循环调用 contains() 的 O(n×m)。

⚠️ 注意事项与进阶建议

  • 大小写敏感性:默认区分大小写。如需忽略大小写,链式调用 .str.to_lowercase() 或启用 strict=False(部分版本需配合正则):

    ~pl.col("filename").str.to_lowercase().str.contains_any(
        df_filter["skip"].str.to_lowercase()
    )
  • 转义特殊字符:若 filter.csv 中含正则元字符(如 .、*、?),需提前转义,否则可能误匹配:

    import re
    escaped_keywords = df_filter["skip"].map_elements(lambda s: re.escape(s))
    df_data.filter(~pl.col("filename").str.contains_any(escaped_keywords))
  • 性能对比关键点
    ❌ 错误示范(低效循环):

    # 每次 filter 都触发一次执行计划重编译 + 全表扫描 → O(m) 次 I/O 和计算
    for keyword in df_filter["skip"]:
        df_data = df_data.filter(~pl.col("filename").str.contains(keyword))

    ✅ 正确范式(单次执行):

    # 一次性构建完整布尔表达式,由 Polars 优化器统一调度
    mask = pl.col("filename").str.contains_any(df_filter["skip"])
    result = df_data.filter(~mask)
  • 替代方案(兼容旧版 Polars)
    若环境版本 < 0.20.0,可手动构造逻辑或(OR)表达式,再取反:

    from functools import reduce
    import operator
    
    patterns = df_filter["skip"].to_list()
    contains_expr = reduce(
        operator.or_,
        (pl.col("filename").str.contains(p) for p in patterns)
    )
    result = df_data.filter(~contains_expr)

? 总结
str.contains_any() 是 Polars 对 SQL LIKE 批量模式匹配的优雅映射——它消除了显式循环,让「关键词黑名单过滤」回归声明式、高性能的数据操作本质。在处理百万级文件名、日志路径或 URL 清洗任务时,该方法可带来数倍至数十倍的性能提升,同时保持代码清晰可维护。务必优先选用此原生接口,而非模拟 SQL 逻辑的手动循环或嵌套表达式。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的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,提供了直观易用的用户界面等等。

1135

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错误的相关内容,可以阅读本专题下面的文章。

2298

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数据库的相关内容,可以阅读本专题下面的文章。

1764

2024.04.07

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

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

588

2024.04.29

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

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

441

2024.04.29

Python WebSocket实时通信与异步服务开发实践
Python WebSocket实时通信与异步服务开发实践

本专题聚焦 Python 在实时通信场景中的开发实践,系统讲解 WebSocket 协议原理、长连接管理、消息推送机制以及异步服务架构设计。内容包括客户端与服务端通信实现、连接稳定性优化、消息队列集成及高并发处理策略。通过完整案例,帮助开发者构建高效稳定的实时通信系统,适用于聊天应用、实时数据推送等场景。

3

2026.03.18

热门下载

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

精品课程

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

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