如何在Python中通过mmap将大文件映射为内存中的数组?

落磊同学_8882

落磊同学_8882

2026-06-17

943人浏览

原创

mmap.mmap() 返回的是类字节对象,不含 dtype、shape 和 stride 等 numpy 数组必需的元信息,故不能直接转为数组;必须通过 np.frombuffer()(显式指定 dtype/count)或 np.memmap()(自动管理布局与文件映射)构建合法 numpy 数组。

如何在python中通过mmap将大文件映射为内存中的数组?

不能直接用 mmap 得到 NumPy 数组,必须配合 numpy.frombuffer() 或 numpy.memmap();否则会报 TypeError: buffer is not compatible with array。

为什么 mmap.mmap() 本身不是 NumPy 数组?

mmap.mmap() 返回的是一个类字节对象(memoryview-like),支持切片和字节级读写,但不带 dtype、shape 或 stride 信息。NumPy 数组需要明确的数据类型和内存布局才能安全解释字节流。

  • 直接传给 np.array(m) 会失败 —— 它尝试拷贝全部内容并推断 dtype,而 mmap 对象不支持 __len__() 在某些平台(如 Windows)上抛错
  • np.asarray(m) 同样失败,原因相同
  • 正确路径只有一条:把 mmap 当作 buffer 传给 np.frombuffer(),并显式指定 dtype 和 count

用 np.frombuffer() 创建只读数组(推荐小到中等文件)

适用于已知 dtype 和总元素数的场景,比如 float32 的二进制记录文件。注意:返回的是只读数组,修改会触发 ValueError: assignment destination is read-only。

import numpy as np
import mmap
<p>with open("data.bin", "rb") as f:
mm = mmap.mmap(f.fileno(), 0, access=mmap.ACCESS_READ)</p><h1>假设是 100 万个 float32</h1><pre class="brush:python;toolbar:false;">arr = np.frombuffer(mm, dtype=np.float32, count=1_000_000)
print(arr.shape)  # (1000000,)
# arr[0] = 1.0  ← 这行会报错
  • count 必须准确,否则越界读取可能静默截断或引发 ValueError
  • 文件大小必须是 dtype.itemsize * count 的整数倍,否则末尾字节被忽略
  • 跨平台安全:Linux/macOS/Windows 都适用

用 np.memmap() 创建可读写、带 shape 的数组(推荐大文件或需 reshape 场景)

np.memmap() 是专为内存映射设计的 NumPy 类,底层仍用 mmap,但自动管理 dtype、offset、shape,并支持赋值(需 mode='r+' 或 'c')。

Li Python Sec Check
Li Python Sec Check

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

下载
import numpy as np
<h1>创建一个 1000x1000 的 float64 memmap(如果文件不存在则新建)</h1><p>arr = np.memmap("big_matrix.dat", dtype="float64", mode="w+", shape=(1000, 1000))</p><h1>写入数据(实际写入磁盘文件)</h1><p>arr[0, :] = np.arange(1000)
arr.flush()  # 强制写回磁盘(非必需,但确保可见)</p><h1>下次打开可读取</h1><p>arr_read = np.memmap("big_matrix.dat", dtype="float64", mode="r", shape=(1000, 1000))
print(arr_read[0, :5])  # [0. 1. 2. 3. 4.]</p>
  • mode='r+' 表示可读可写,但文件必须已存在且大小匹配;mode='w+' 会清空或新建文件
  • shape 和 dtype 一起决定了所需文件大小,不匹配会报 ValueError: size of the array must match the size of the file
  • 修改后不调用 .flush() 也可能被写入,但依赖 OS 缓存策略,生产环境建议显式 flush

常见坑与绕过技巧

遇到 OSError: [Errno 22] Invalid argument 或数组全为零,大概率是 mmap 参数或 dtype 不对齐。

  • Windows 上 length=0 必须配合 access=mmap.ACCESS_READ 或 ACCESS_WRITE,不能用 ACCESS_COPY
  • struct 打包的混合类型(如 int32 + float64)不能直接用 frombuffer() 解释,得先用 struct.unpack() 或 np.fromfile() 配合 dtype 描述符
  • 想映射超大文件(>2GB)在 32 位 Python 下会失败,确认用的是 64 位解释器
  • np.memmap() 的 offset 参数可用于跳过文件头(如 512 字节 header),但 offset 必须是 dtype.itemsize 的整数倍,否则报错

真正麻烦的地方不在映射本身,而在 dtype 对齐、文件边界和跨平台 mmap 行为差异 —— 调试时先用 os.path.getsize() 和 dtype.itemsize 手动验算字节数是否匹配。

Python免费学习笔记(深入):立即使用
在学习笔记中,你将探索 Python 的核心概念和高级技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

python

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

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1671

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4204

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1669

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

24457

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2987

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3007

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1163

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2343

5

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程