Python内存映射文件读写方式
我就废话不多说了,还是直接看代码吧!
import os
import time
import mmap
filename = 'test.txt'
#如果不存在,创建。
if not os.path.exists(filename):
open(filename, 'w')
print(os.path.isdir(filename))
if os.path.isfile(filename):
print(time.ctime(os.path.getctime(filename)))
fd = os.open(filename, os.O_RDWR)
m = mmap.mmap(fd, 50, access=mmap.ACCESS_WRITE) # 1024字节的文件。
m.seek(2)
buf1 = bytes(b'Zhang')
m[2:len(buf1)+2] = buf1
buf2 = b'Phil'
m.seek(20) # 定位写入的位置。
m.write(buf2) # 写入字节数据。
m.close()
fd = os.open(filename, os.O_RDWR)
m = mmap.mmap(fd, 50, access=mmap.ACCESS_READ)
m.seek(20)
buf3 = m.read(len(buf2))
m.close()
print(list(buf3))
操作后的test.txt文件内容:
Zhang Phil
补充知识:Python对二进制文件做内存映射,高效优雅地对内容随机访问
读写二进制文件还在使用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:11]=b'Hello World'#改变一个切片
m.close()
with open('data','rb') as f:#验证改变已经生效
print(f.read(11))
b'Hello World'
由于mmap()返回的mmap对象也可以当作上下文管理器使用,在这种情况下,底层的文件会自动关闭。
with memory_map('data') as m:
print(len(m))
print(m[0:11])