展开菜单
首页 精品内容 本月促销 装机必备 Windows PDF教程 macOS软件 IOS软件 Android AI 专题
全部分类

当前位置:

首页 > 编程开发 > 如何在Python中使用Pandas读取HDF5格式的高性能大型数据集?

如何在Python中使用Pandas读取HDF5格式的高性能大型数据集?

读取HDF5时,先用h5py探明结构避免键名错误。pd.read_hdf()仅支持PyTables格式,需写入时指定format='table'和data_columns。非Pandas数据可用h5py切片读取转换。写入推荐format='table'、data_columns及blosc压缩以提升后续读取效率。

本文目录

    先说说各位在使用Pandas读取HDF5文件时,最让人头疼的两个问题:要么是内存直接爆掉,要么是收到一个莫名其妙的`KeyError: 'No object named … in the file'`。

    问题出在HDF5文件的结构上。它不是一个“一张表对应一个文件”的简单玩意儿,而是一个像文件系统一样的容器,里头可以存放多个数据集(就是代码里的`key`)。而且,它默认不会告诉你里面到底有哪些东西。你直接调用`pd.read_hdf('data.h5')`,Pandas会尝试去读取一个叫`'/df'`或`'table'`的默认键。但现实是,很多生产环境下的HDF5文件,根本不会按这个约定来命名。

    这时候,正确的做法是先摸清文件结构,用`h5py`来探路:

    import h5py
    with h5py.File('data.h5', 'r') as f:
    print(list(f.keys())) # 看看顶层有哪些group
    # 如果还有嵌套,比如 '/processed/2023Q1/sales',就需要写上完整路径

    另外,还有两个常见坑需要留意:

    • pd.read_hdf() 只能读取它自己用 pd.to_hdf() 写进去的 Table 或 Fixed 格式。如果这个HDF5是用 h5py、MATLAB 或者别的什么工具生成的原始数据集,那它就没辙了。
    • 键名里如果带着斜杠(比如 '/raw/feature_matrix'),引用时一定要加引号,并且开头的那个 / 不能省略。
    • 一个容易忽略的安全点:处理大文件时,务必用 mode='r'(只读模式),不小心用了 'a' 模式,可能会触发一些意想不到的写入操作,甚至把文件锁死。

    如何用 pd.read_hdf() 高效读取子集,避免全量加载

    HDF5 支持按行过滤和按列选取,但这有个前提:只对一种格式生效——format='table'(也就是 PyTables 格式)。如果当初写入时用的是 format='fixed',那这些参数就全白费了,Pandas还是会老老实实把整个文件砸进内存里。

    读取时,有几个关键参数组合值得记住:

    • where:传入查询条件,比如 "date > '2023-01-01' & category == 'A'"。但要注意,前提是这些列在写入时已经被索引了(通过 data_columns=['date', 'category'] 指定)。
    • columns:指定你真正需要的列名列表,比如 ['user_id', 'event_time'],这能显著减少内存占用,属于“精准打击”。
    • start/stop:按行号切片读取。对于没有索引的大宽表,这个方法比 where 更快,但它没法做条件过滤。
    • iterator=True + chunksize=10000:返回一个 HDFStore 迭代器。如果你要处理的数据流非常大,这个组合能帮你实现流式处理,一点一点消化。

    举个例子,带条件加列筛选,清爽又高效:

    df = pd.read_hdf(
    'data.h5',
    key='/sales/table',
    where="region == 'CN' and revenue > 1000",
    columns=['order_id', 'revenue', 'ts'],
    )

    用 h5py 手动读取原始 dataset 后转为 DataFrame 的适用场景

    当你的HDF5文件是科学计算常用的格式(比如直接存了NumPy数组),或者是由非Pandas工具生成的,那么pd.read_hdf()就会彻底失效。这时候,必须请出底层的 h5py 来亲自操刀。

    典型的操作流程是这样:

    • 用 f[key] 拿到数据集对象,再调用 [...] 触发读取。特别提醒:[...] 会一次性把数据全量加载到内存,如果文件很大,务必小心。
    • 数据太大怎么办?用 f[key][start:stop] 切片读取,然后分批拼成DataFrame,这是避免OOM的常规操作。
    • 列名需要手动构造:pd.DataFrame(data, columns=col_names)。这些列名通常藏在同一个group的 attrs 里,或者单独有个group存放。
    • 时间戳字段经常是 int64 类型(纳秒级的Unix时间戳),需要手动转换:pd.to_datetime(..., unit='ns')。

    看一个安全切片读取的例子:

    import h5py
    import numpy as np

    with h5py.File('sensor.h5', 'r') as f:
    dset = f['/measurements']
    # 分块读取,避免内存爆炸
    chunk = dset[0:50000] # shape (50000, 12)
    df = pd.DataFrame(chunk, columns=['temp', 'hum', 'vib_x', ...])

    写入时就该注意的三点,决定后续读取效率

    读取慢,或者功能受限,90%的锅都能甩到写入时的配置上。关键参数必须在 to_hdf() 时明确指定,不能偷懒:

    • format='table':这是开启查询、列选取、增量追加等高级功能的大门。而 format='fixed' 虽然写入快,但功能受限,后续只能全量读取。
    • data_columns:把你未来要频繁过滤的字段(比如 date, user_id)都加到这个列表里。否则,where 参数就是个摆设。
    • complib='blosc' + complevel=9:这对组合对数值型大数据的压缩效率高,解压速度也快。相比之下,zlib 在Pandas HDF5里的支持表现就比较一般。

    很多新手图省事,写出来往往是这样的:

    # ❌ 不推荐:没指定 format 和 data_columns
    df.to_hdf('out.h5', 'data')

    但真正懂行的人,会这么写:

    # ✅ 推荐:可过滤、可选列、高压缩
    df.to_hdf(
    'out.h5',
    'data',
    format='table',
    data_columns=['timestamp', 'device_id'],
    complib='blosc',
    complevel=9
    )

    说到底,真正难的不是怎么读,而是先搞清楚这个HDF5文件里到底存了什么、是怎么存的。花两分钟时间,用 h5py 看一眼结构,远比你在那里反复试错 key 和 where 参数要省事得多。这,才是效率的根源。

    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    编程开发 Python
    相关文章 更多
    精品专题 更多
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    macOS软件

    正软商城macOS软件专区,精选适用于Mac电脑的办公、设计、影音、效率、开发和系统工具,提供软件功能介绍、macOS兼容版本、正版授权及购买下载服务。

    IOS软件

    正软商城iOS软件专区,精选适用于iPhone和iPad的办公、学习、影音、设计、效率及AI应用,提供功能介绍、适用设备、系统要求和正版获取方式等信息。

    AI

    正软商城AI软件专区,汇集AI写作、AI绘画、AI视频、AI办公、AI编程、AI翻译、智能客服和数据分析等人工智能工具,提供功能介绍、适用平台、收费方式及正版购买信息。

    Mac软件 更多
    IntelliJ IDEA macOS版
    IntelliJ IDEA macOS版

    IntelliJ IDEA 是 JetBrains 推出的专业集成开发环境,重点服务于 Java、Kotlin 与 JVM 项目开发。软件集成代码补全、静态检查、重构、调试、测试、版本控制及构建工具,并可借助插件扩展语言和框架支持。

    PyCharm macOS版
    PyCharm macOS版

    JetBrains PyCharm 是面向 Python 开发者的集成开发环境,提供智能补全、代码检查、重构、调试、测试、版本控制及虚拟环境管理功能。Mac 版兼容 Intel 与 Apple 芯片,核心功能可免费使用。

    WebStorm macOS版
    WebStorm macOS版

    JetBrains WebStorm 是面向 JavaScript、TypeScript 和现代 Web 技术的专业集成开发环境,可智能理解项目结构,提供代码补全、重构、错误检查、运行调试、测试及 Git 工具。

    After Effects macOS版
    After Effects macOS版

    Adobe After Effects 是面向动态图形、影视特效与视频合成的专业创作软件。它提供关键帧动画、遮罩抠像、运动跟踪、三维空间、文字动画和表达式等工具,并可与 Premiere Pro、Photoshop、Illustrator

    Audition macOS版
    Audition macOS版

    Adobe Audition 是面向播客、影视后期、音乐制作及内容创作者的专业音频工作站,提供波形精修、多轨录制与混音、频谱编辑、降噪修复、响度匹配和效果处理工具,并可与 Adobe Premiere 协同完成视频声音制作。

    Bridge macOS版
    Bridge macOS版

    Adobe Bridge 是面向摄影师、设计师及内容团队的免费数字资产管理工具,可集中预览照片、视频和 Adobe 项目文件,并利用评级、标签、关键词、元数据、筛选与收藏集完成分类检索。它还支持批量重命名、格式导出、相机导入及 Camera

    Illustrator macOS版
    Illustrator macOS版

    Adobe Illustrator 是 Adobe 推出的专业矢量图形设计软件,适合在 Mac 上制作标志、图标、插画、包装、信息图及印刷版面。它以路径和锚点构建可无损缩放的图形,并提供钢笔、形状、文字、图像描摹、渐变、画板及生成式功能。

    InDesign macOS版
    InDesign macOS版

    Adobe InDesign 是面向印刷品与数字出版物的专业版面设计软件,可在 Mac 上制作书籍、杂志、宣传册、海报、报告及交互式文档。它提供精细的文字样式、网格、母版、长文档管理和印前输出能力。

    Lightroom Classic macOS版
    Lightroom Classic macOS版

    Adobe Lightroom Classic 是面向桌面摄影工作流的照片管理与后期处理软件,可将原始照片保存在本地硬盘,通过目录、关键词、评分和收藏夹高效整理图库,并提供 RAW 冲印、蒙版、镜头校正、批量同步及多格式导出等功能。

    Media Encoder macOS版
    Media Encoder macOS版

    Adobe Media Encoder 是 Adobe 推出的专业音视频编码与转码工具,可通过队列、预设和监视文件夹批量完成格式转换、代理文件创建及多平台成片输出,并与 Premiere Pro、After Effects 等软件紧密协作。

    Photoshop macOS版
    Photoshop macOS版

    Adobe Photoshop 是面向摄影师、设计师和内容创作者的专业图像处理软件,提供图层、蒙版、选区、修复、调色、文字排版、智能对象及生成式编辑等能力。Mac 版支持 Intel 与 Apple Silicon 处理器。

    Premiere Pro macOS版
    Premiere Pro macOS版

    Adobe Premiere Pro 是面向专业视频制作的非线性剪辑软件,提供多轨时间线、调色、音频处理、字幕、代理工作流和多格式输出能力,并可与 After Effects、Audition、Photoshop 及 Frame.io 协同

    WINDOWS 更多
    IntelliJ IDEA Windows版
    IntelliJ IDEA Windows版

    JetBrains IntelliJ IDEA 是以 Java、Kotlin 和 JVM 项目为核心的专业集成开发环境,提供智能补全、代码检查、安全重构、调试、测试、版本控制及构建工具集成。统一发行版的核心功能可免费用于商业和非商业项目,

    PhpStorm Windows版
    PhpStorm Windows版

    PhpStorm 是 JetBrains 面向 PHP 与现代 Web 开发打造的专业集成开发环境,提供代码补全、静态检查、重构、调试、测试、数据库、版本控制及前端语言支持。Windows 用户可直接下载安装程序,

    PyCharm Windows版
    PyCharm Windows版

    PyCharm 是 JetBrains 推出的 Python 集成开发环境,提供智能代码补全、错误检查、重构、调试、测试、Git、终端及 Jupyter Notebook 支持。统一版安装后可体验 30 天 Pro 功能,

    WebStorm Windows版
    WebStorm Windows版

    WebStorm 是 JetBrains 面向 JavaScript、TypeScript 和现代 Web 技术打造的集成开发环境。它提供项目级代码分析、智能补全、安全重构、导航搜索,并集成调试、测试、终端、Git 与常用前端工具,

    Photoshop Windows版
    Photoshop Windows版

    Adobe Photoshop 是面向摄影师、设计师和内容创作者的专业图像处理工具,可在 Windows 上完成照片修饰、选区抠图、图层合成、文字排版、绘画及生成式编辑。软件采用订阅授权,由 Creative Cloud 管理安装、更新与许

    InDesign Windows版
    InDesign Windows版

    Adobe InDesign 是面向印刷与数字出版的专业页面排版工具,可制作书籍、杂志、宣传册、海报、交互式 PDF 和电子书。它提供精细的文字样式、多页面管理、印前检查及多格式输出能力,并能与 Photoshop、Illustrator

    Illustrator Windows版
    Illustrator Windows版

    Adobe Illustrator 是用于绘制矢量图形、插画、图标、徽标和版式的专业设计工具。它以路径和锚点为核心,作品放大后仍能保持清晰,并提供画板、文字、渐变、图像描摹及精确变换等功能,适合品牌视觉、印刷物料、数字界面和商业插画创作。

    Bridge Windows版
    Bridge Windows版

    Adobe Bridge 是 Adobe 推出的创意资产管理工具,可集中预览、分类和查找照片、视频、PDF及多种设计文件,并支持元数据、关键词、评级、批量重命名和格式导出。它可与 Photoshop、Illustrator 等应用衔接,

    Lightroom Classic Windows版
    Lightroom Classic Windows版

    Adobe Lightroom Classic 是面向 Windows 桌面摄影工作流的照片管理与无损编辑工具,可通过目录集中整理大型图库,并完成 RAW 显影、局部蒙版、批量调整、打印和多格式导出。它强调本地文件管理与精细控制,

    Acrobat Pro Windows版
    Acrobat Pro Windows版

    Adobe Acrobat Pro 是面向专业 PDF 工作的综合解决方案,可在 Windows 中创建、编辑、转换、比较、整理和保护文档,并支持表单、电子签名、扫描件文字识别及敏感信息永久遮盖。订阅还包含网页、移动服务和云端协作能力,

    After Effects Windows版
    After Effects Windows版

    Adobe After Effects 是面向动态图形、动画与影视视觉特效制作的专业工具,可完成关键帧动画、图层合成、遮罩跟踪、抠像、文字动画和二维三维元素整合。Windows 版通过 Adobe Creative Cloud 安装及更新,

    Media Encoder Windows版
    Media Encoder Windows版

    Adobe Media Encoder 是 Adobe 面向视频与音频制作流程提供的专业编码和转码工具,可批量处理媒体文件、创建代理、套用输出预设并监控文件夹自动执行任务。它与 Premiere、After Effects、Audition