Python內(nèi)存映射文件讀寫(xiě)方式
我就廢話不多說(shuō)了,還是直接看代碼吧!
import osimport timeimport mmap filename = ’test.txt’ #如果不存在,創(chuàng)建。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字節(jié)的文件。m.seek(2) buf1 = bytes(b’Zhang’)m[2:len(buf1)+2] = buf1 buf2 = b’Phil’ m.seek(20) # 定位寫(xiě)入的位置。m.write(buf2) # 寫(xiě)入字節(jié)數(shù)據(jù)。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文件內(nèi)容:
Zhang Phil
補(bǔ)充知識(shí):Python對(duì)二進(jìn)制文件做內(nèi)存映射,高效優(yōu)雅地對(duì)內(nèi)容隨機(jī)訪問(wèn)
讀寫(xiě)二進(jìn)制文件還在使用open函數(shù)?
各種組合seek()、read()和write()累不累?
使用 mmap 模塊實(shí)現(xiàn)對(duì)文件的內(nèi)存映射,讓我們讀寫(xiě)二進(jìn)制文件像操作數(shù)組一樣高效優(yōu)雅。
先給出一個(gè)實(shí)用函數(shù),用來(lái)演示如何打開(kāi)一個(gè)文件并對(duì)它進(jìn)行內(nèi)存映射操作。
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)
要使用這個(gè)函數(shù),需要先準(zhǔn)備一個(gè)已經(jīng)創(chuàng)建好的文件并為之填充一些數(shù)據(jù)。
>>> size = 1000000>>> with open(’data’,’wb’) as f: f.seek(size-1) f.write(b’x00’)
接著我們可以使用memory_map()函數(shù)對(duì)文件內(nèi)容做內(nèi)存映射,它返回的mmap對(duì)象可以讓我們像操作數(shù)組一樣讀寫(xiě)二進(jìn)制文件!
>>> m = memory_map(’data’)>>> len(m)1000000>>> m[:10]b’x00x00x00x00x00x00x00x00x00x00’>>> m[0]0>>> m[0:11]=b’Hello World’#改變一個(gè)切片>>> m.close()>>> with open(’data’,’rb’) as f:#驗(yàn)證改變已經(jīng)生效 print(f.read(11)) b’Hello World’
由于mmap()返回的mmap對(duì)象也可以當(dāng)作上下文管理器使用,在這種情況下,底層的文件會(huì)自動(dòng)關(guān)閉。
>>> with memory_map(’data’) as m: print(len(m)) print(m[0:11]) 1000000b’Hello World’
默認(rèn)情況下memory_map()函數(shù)打開(kāi)的文件既可以讀,也可以寫(xiě)。對(duì)數(shù)據(jù)的任何修改都會(huì)拷貝回原始的文件中。
如果需要只讀訪問(wèn),可以為access參數(shù)提供mmap.ACCESS_READ值。
m = memory_map(filename, mmap.ACCESS_READ)
如果只想要在本地修改數(shù)據(jù),并不想將這些修改寫(xiě)回到原始文件中,可以使用mmap.ACCESS_COPY參數(shù)。
m = memory_map(filename, mmap.ACCESS_COPY)
總結(jié):
通過(guò)mmap將文件映射到內(nèi)存之后,我們可以高效并優(yōu)雅地對(duì)文件的內(nèi)容進(jìn)行隨機(jī)訪問(wèn)。
與其打開(kāi)文件后通過(guò)組合各種seek()、read()和write()調(diào)用來(lái)訪問(wèn),不如簡(jiǎn)單將文件映射到內(nèi)存,然后通過(guò)切片操作來(lái)訪問(wèn)數(shù)據(jù)。
需要強(qiáng)調(diào)的是,對(duì)某個(gè)文件進(jìn)行內(nèi)存映射并不會(huì)導(dǎo)致將整個(gè)文件讀到內(nèi)存中。也就是說(shuō),文件并不會(huì)拷貝到某種內(nèi)存緩沖區(qū)或數(shù)組上。相反,操作系統(tǒng)只是為文件內(nèi)容保留一段虛擬內(nèi)存而已。
當(dāng)訪問(wèn)文件的不同區(qū)域時(shí),文件的這些區(qū)域?qū)⒈蛔x取并按照需要映射到內(nèi)存區(qū)域中。但是,文件中從未訪問(wèn)過(guò)的部分會(huì)簡(jiǎn)單地留在磁盤(pán)上。這一切都是以透明的方式在幕后完成的。
如果有多個(gè)Python解釋器對(duì)同一個(gè)文件做了內(nèi)存映射,得到的mmap對(duì)象可以用來(lái)在解釋器之間交換數(shù)據(jù)。也就是說(shuō),所有的解釋器可以同時(shí)讀/寫(xiě)數(shù)據(jù),在一個(gè)解釋器中對(duì)數(shù)據(jù)做出的修改會(huì)自動(dòng)反映到其他解釋器上。
雖然還需要一些額外的步驟來(lái)處理同步問(wèn)題,但是有時(shí)候可以使用這種方法作為通過(guò)管道或者socket傳輸數(shù)據(jù)的替代方案。
以上這篇Python內(nèi)存映射文件讀寫(xiě)方式就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
1. Python網(wǎng)絡(luò)編程之ZeroMQ知識(shí)總結(jié)2. Python查找算法之分塊查找算法的實(shí)現(xiàn)3. python基于tkinter點(diǎn)擊按鈕實(shí)現(xiàn)圖片的切換4. .NET 中配置從xml轉(zhuǎn)向json方法示例詳解5. python實(shí)現(xiàn)讀取類別頻數(shù)數(shù)據(jù)畫(huà)水平條形圖案例6. python之cur.fetchall與cur.fetchone提取數(shù)據(jù)并統(tǒng)計(jì)處理操作7. Python TestSuite生成測(cè)試報(bào)告過(guò)程解析8. 解決AJAX返回狀態(tài)200沒(méi)有調(diào)用success的問(wèn)題9. JSP之表單提交get和post的區(qū)別詳解及實(shí)例10. 低版本IE正常運(yùn)行HTML5+CSS3網(wǎng)站的3種解決方案
