0

0

Python pyarrow 的 dataset API 实践

冰川箭仙

冰川箭仙

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

|

1019人浏览过

|

来源于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免费学习笔记(深入)”;

Bardeen AI
Bardeen AI

使用AI自动执行人工任务

下载
  • 必须导入 import pyarrow.compute as pc
  • 写法只能是 pc.field("col") == "val"pc.greater(pc.field("ts"), pc.scalar("2023-01-01"))
  • 字符串匹配只能用 pc.match_substring(),不是 LIKEpc.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聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的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,提供了直观易用的用户界面等等。

1047

2023.10.12

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

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

338

2023.10.27

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

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

379

2024.02.23

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

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

1862

2024.03.06

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

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

378

2024.03.06

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

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

1415

2024.04.07

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

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

585

2024.04.29

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

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

437

2024.04.29

batoto漫画官网入口与网页版访问指南
batoto漫画官网入口与网页版访问指南

本专题系统整理batoto漫画官方网站最新可用入口,涵盖最新官网地址、网页版登录页面及防走失访问方式说明,帮助用户快速找到batoto漫画官方平台,稳定在线阅读各类漫画内容。

127

2026.02.25

热门下载

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

精品课程

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

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.7万人学习

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

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