读写二进制文件还在使用open函数?
各种组合seek()、read()和write()累不累?
使用 mmap 模块实现对文件的内存映射,让我们读写二进制文件像操作数组一样高效优雅。
先给出一个实用函数,用来演示如何打开一个文件并对它进行内存映射操作。
def memory_map(filename, access=mmap.ACCESS_WRITE):
size = os.path.getsize(filename)
fd = os.open(filename, os.O_RDWR)
return mmap.mmap(fd, size, access=access)
要使用这个函数,需要先准备一个已经创建好的文件并为之填充一些数据。
>>> size = 1000000
>>> with open('data','wb') as f:
f.seek(size-1)
f.write(b'\x00')
接着我们可以使用memory_map()函数对文件内容做内存映射,它返回的mmap对象可以让我们像操作数组一样读写二进制文件!
>>> m = memory_map('data')
>>> len(m)
1000000
>>> m[:10]
b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00'
>>> m[0]
0
>>> m[0:11]=b'Hel