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

Python pyarrow 的 dataset API 实践

阿雪姑娘_1436

阿雪姑娘_1436

发布时间:2026-02-25 20:10:03

|

1059人浏览过

|

来源于php中文网

原创

dataset.read_table()读空的主因是目录格式不匹配:默认仅识别Hive分区路径或含_metadata文件,裸Parquet文件需显式设format="parquet"或partitioning=False;filter须用pyarrow.compute表达式,如pc.match_substring(pc.field("name"),"abc")。

python pyarrow 的 dataset api 实践

为什么 dataset.read_table() 读出来是空的

常见现象:用 dataset.read_table() 加载 Parquet 目录后,len(table) 是 0,但文件明明存在、用 pyarrow.parquet.read_table() 单个文件却能读出数据。

根本原因是:Dataset 默认只识别符合 Hive 分区路径格式(如 year=2023/month=04/)或含 _metadata/_common_metadata 的目录;裸 parquet 文件列表不被自动发现。

  • 确认是否用了 format="parquet" 显式指定(默认会尝试推断,但容易失败)
  • 检查路径下是否有 _metadata 文件;没有就加 partitioning=False 或显式传 partitioning="hive"
  • 如果只是几个独立 .parquet 文件,别用 dataset,直接用 pyarrow.parquet.read_table([f1, f2])

filter 参数写不对,完全不生效

dataset.filter 不是 SQL WHERE,它是基于 Arrow 计算表达式的逻辑树,不支持字符串模糊匹配、函数调用或列别名。

典型错误:ds.filter("col LIKE '%abc%'") 或 ds.filter(pc.field("col") == "val")(漏掉 pc 模块导入或写法错)。

立即学习“Python免费学习笔记(深入)”;

macOS UI 自动化技能包 —— 用 Python 控制你的 Mac
macOS UI 自动化技能包 —— 用 Python 控制你的 Mac

让 macOS 自动化像说话一样简单 —— 用代码控制你的 Mac,无需学习 AppleScript 系统要求 macOS 14+ (Sonoma) Python 3.9+ 需要开启:辅助功能 + 屏幕录制权限

下载
  • 必须导入 import pyarrow.compute as pc
  • 写法只能是 pc.field("col") == "val"、pc.greater(pc.field("ts"), pc.scalar("2023-01-01"))
  • 字符串匹配只能用 pc.match_substring(),不是 LIKE:pc.match_substring(pc.field("name"), "abc")
  • filter 在 scan 阶段下推,但不会改变 schema;若字段不存在,运行时报 KeyError,不是静默跳过

scan() + to_table() 比 read_table() 慢很多

表面看都是读成 Table,但行为完全不同:read_table() 是“一次性全量加载+内存聚合”,scan().to_table() 是“先建执行计划+逐 batch 扫描+最后合并”,中间多一层计算图调度开销。

尤其在小数据集(read_table() 更快也更省内存。

  • 用 scan() 的真实价值在于:需要复用 scan 对象做多次不同 filter/project、或接 to_reader() 流式消费、或配合 fragments 做自定义分片
  • 如果只是“读完就处理”,优先用 read_table(filter=..., columns=[...])
  • read_table(use_threads=True) 默认已开并行,不用额外配;但 scan() 的线程数需通过 use_threads 显式传给 to_table()

写入 dataset 时分区字段没进数据,反而变成目录名

分区字段不会自动保留在每张表的 column 里 —— 这是 Dataset 的设计约定:分区信息只存路径,不冗余进数据本身。

如果你需要分区列也在表中(比如后续要做 groupby 或 join),必须手动加回来。

  • 写入时用 partitioning=pa.dataset.partitioning(...) 只控制目录结构,不影响 table schema
  • 读取后想恢复分区字段,得用 ds.to_table(columns=["a", "b"], fragments=...) + 手动从 fragment.path 提取值再 concat
  • 更稳妥的做法:写入前把分区列作为普通列保留,再用 existing_data_behavior="delete_matching" 配合 partitioning 控制目录,避免丢失语义

分区不是魔法,它省的是扫描 IO,不是数据建模成本。路径解析、类型对齐、空分区处理,都得自己兜底。

热门AI工具

更多
DeepSeek

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

花生AI
花生AI Hot

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

WorkBuddy

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

超级麦吉

一款通过数字员工协作完成调研、数据分析与文档制作的AI办公平台。

蛙蛙写作

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

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

豆包大模型

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

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

立刻MV
立刻MV Hot

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

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

4643

2023.10.12

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

951

2023.10.27

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

1209

2024.02.23

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

6621

2024.03.06

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

3363

2024.03.06

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

6600

2024.04.07

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

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

9081

2024.04.29

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

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

1230

2024.04.29

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

60

2026.10.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
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