讀寫文件

讀寫文件是最常見的IO操作锚沸。Python內(nèi)置了讀寫文件的函數(shù),用法和C是兼容的涕癣。

讀寫文件前哗蜈,我們先必須了解一下,在磁盤上讀寫文件的功能都是由操作系統(tǒng)提供的坠韩,現(xiàn)代操作系統(tǒng)不允許普通的程序直接操作磁盤距潘,所以,讀寫文件就是請求操作系統(tǒng)打開一個文件對象(通常稱為文件描述符)同眯,然后绽昼,通過操作系統(tǒng)提供的接口從這個文件對象中讀取數(shù)據(jù)(讀文件)唯鸭,或者把數(shù)據(jù)寫入這個文件對象(寫文件)须蜗。

讀文件

要以讀文件的模式打開一個文件對象,使用Python內(nèi)置的open()函數(shù)目溉,傳入文件名和標(biāo)示符:

>>> f = open('/Users/michael/test.txt', 'r')

標(biāo)示符'r'表示讀明肮,這樣,我們就成功地打開了一個文件缭付。

如果文件不存在柿估,open()函數(shù)就會拋出一個IOError的錯誤,并且給出錯誤碼和詳細(xì)的信息告訴你文件不存在:

>>> f=open('/Users/michael/notfound.txt', 'r')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
FileNotFoundError: [Errno 2] No such file or directory: '/Users/Michael/notfound.txt'

如果文件打開成功陷猫,接下來秫舌,調(diào)用read()方法可以一次讀取文件的全部內(nèi)容的妖,Python把內(nèi)容讀到內(nèi)存,用一個str對象表示:

>>> f.read()
'Hello, world!'

最后一步是調(diào)用close()方法關(guān)閉文件足陨。文件使用完畢后必須關(guān)閉嫂粟,因?yàn)槲募ο髸加貌僮飨到y(tǒng)的資源,并且操作系統(tǒng)同一時間能打開的文件數(shù)量也是有限的:

>>> f.close()

由于文件讀寫時都有可能產(chǎn)生IOError墨缘,一旦出錯星虹,后面的f.close()就不會調(diào)用。所以镊讼,為了保證無論是否出錯都能正確地關(guān)閉文件宽涌,我們可以使用try ... finally來實(shí)現(xiàn):

try:
    f = open('/path/to/file', 'r')
    print(f.read())
finally:
    if f:
        f.close()

但是每次都這么寫實(shí)在太繁瑣,所以蝶棋,Python引入了with語句來自動幫我們調(diào)用close()方法:

with open('/path/to/file', 'r') as f:
    print(f.read())

這和前面的try ... finally是一樣的卸亮,但是代碼更佳簡潔,并且不必調(diào)用f.close()方法玩裙。

調(diào)用read()會一次性讀取文件的全部內(nèi)容嫡良,如果文件有10G,內(nèi)存就爆了献酗,所以寝受,要保險起見,可以反復(fù)調(diào)用read(size)方法罕偎,每次最多讀取size個字節(jié)的內(nèi)容很澄。另外,調(diào)用readline()可以每次讀取一行內(nèi)容颜及,調(diào)用readlines()一次讀取所有內(nèi)容并按行返回list甩苛。因此,要根據(jù)需要決定怎么調(diào)用俏站。

如果文件很小讯蒲,read()一次性讀取最方便;如果不能確定文件大小肄扎,反復(fù)調(diào)用read(size)比較保險墨林;如果是配置文件,調(diào)用readlines()最方便:

for line in f.readlines():
    print(line.strip()) # 把末尾的'\n'刪掉

file-like Object

open()函數(shù)返回的這種有個read()方法的對象犯祠,在Python中統(tǒng)稱為file-like Object旭等。除了file外,還可以是內(nèi)存的字節(jié)流衡载,網(wǎng)絡(luò)流搔耕,自定義流等等。file-like Object不要求從特定類繼承痰娱,只要寫個read()方法就行弃榨。

StringIO就是在內(nèi)存中創(chuàng)建的file-like Object菩收,常用作臨時緩沖。

二進(jìn)制文件

前面講的默認(rèn)都是讀取文本文件鲸睛,并且是UTF-8編碼的文本文件坛梁。要讀取二進(jìn)制文件,比如圖片腊凶、視頻等等划咐,用'rb'模式打開文件即可:

>>> f = open('/Users/michael/test.jpg', 'rb')
>>> f.read()
b'\xff\xd8\xff\xe1\x00\x18Exif\x00\x00...' # 十六進(jìn)制表示的字節(jié)

字符編碼

要讀取非UTF-8編碼的文本文件,需要給open()函數(shù)傳入encoding參數(shù)钧萍,例如褐缠,讀取GBK編碼的文件:

>>> f = open('/Users/michael/gbk.txt', 'r', encoding='gbk')
>>> f.read()
'測試'

遇到有些編碼不規(guī)范的文件,你可能會遇到UnicodeDecodeError风瘦,因?yàn)樵谖谋疚募锌赡軍A雜了一些非法編碼的字符队魏。遇到這種情況,open()函數(shù)還接收一個errors參數(shù)万搔,表示如果遇到編碼錯誤后如何處理胡桨。最簡單的方式是直接忽略:

>>> f = open('/Users/michael/gbk.txt', 'r', encoding='gbk', errors='ignore')

寫文件

寫文件和讀文件是一樣的,唯一區(qū)別是調(diào)用open()函數(shù)時瞬雹,傳入標(biāo)識符'w'或者'wb'表示寫文本文件或?qū)懚M(jìn)制文件:

>>> f = open('/Users/michael/test.txt', 'w')
>>> f.write('Hello, world!')
>>> f.close()

你可以反復(fù)調(diào)用write()來寫入文件昧谊,但是務(wù)必要調(diào)用f.close()來關(guān)閉文件。當(dāng)我們寫文件時酗捌,操作系統(tǒng)往往不會立刻把數(shù)據(jù)寫入磁盤呢诬,而是放到內(nèi)存緩存起來,空閑的時候再慢慢寫入胖缤。只有調(diào)用close()方法時尚镰,操作系統(tǒng)才保證把沒有寫入的數(shù)據(jù)全部寫入磁盤。忘記調(diào)用close()的后果是數(shù)據(jù)可能只寫了一部分到磁盤哪廓,剩下的丟失了狗唉。所以,還是用with語句來得保險:

with open('/Users/michael/test.txt', 'w') as f:
    f.write('Hello, world!')

要寫入特定編碼的文本文件涡真,請給open()函數(shù)傳入encoding參數(shù)分俯,將字符串自動轉(zhuǎn)換成指定編碼。

細(xì)心的童鞋會發(fā)現(xiàn)综膀,以'w'模式寫入文件時澳迫,如果文件已存在局齿,會直接覆蓋(相當(dāng)于刪掉后新寫入一個文件)剧劝。如果我們希望追加到文件末尾怎么辦?可以傳入'a'以追加(append)模式寫入抓歼。

所有模式的定義及含義可以參考Python的官方文檔讥此。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末拢锹,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子萄喳,更是在濱河造成了極大的恐慌卒稳,老刑警劉巖,帶你破解...
    沈念sama閱讀 219,039評論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件他巨,死亡現(xiàn)場離奇詭異充坑,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)染突,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,426評論 3 395
  • 文/潘曉璐 我一進(jìn)店門捻爷,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人份企,你說我怎么就攤上這事也榄。” “怎么了司志?”我有些...
    開封第一講書人閱讀 165,417評論 0 356
  • 文/不壞的土叔 我叫張陵甜紫,是天一觀的道長。 經(jīng)常有香客問我骂远,道長囚霸,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,868評論 1 295
  • 正文 為了忘掉前任激才,我火速辦了婚禮邮辽,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘贸营。我一直安慰自己吨述,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,892評論 6 392
  • 文/花漫 我一把揭開白布钞脂。 她就那樣靜靜地躺著揣云,像睡著了一般。 火紅的嫁衣襯著肌膚如雪冰啃。 梳的紋絲不亂的頭發(fā)上邓夕,一...
    開封第一講書人閱讀 51,692評論 1 305
  • 那天,我揣著相機(jī)與錄音阎毅,去河邊找鬼焚刚。 笑死,一個胖子當(dāng)著我的面吹牛扇调,可吹牛的內(nèi)容都是我干的矿咕。 我是一名探鬼主播,決...
    沈念sama閱讀 40,416評論 3 419
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼碳柱!你這毒婦竟也來了捡絮?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,326評論 0 276
  • 序言:老撾萬榮一對情侶失蹤莲镣,失蹤者是張志新(化名)和其女友劉穎福稳,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體瑞侮,經(jīng)...
    沈念sama閱讀 45,782評論 1 316
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡的圆,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,957評論 3 337
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了半火。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片略板。...
    茶點(diǎn)故事閱讀 40,102評論 1 350
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖慈缔,靈堂內(nèi)的尸體忽然破棺而出叮称,到底是詐尸還是另有隱情,我是刑警寧澤藐鹤,帶...
    沈念sama閱讀 35,790評論 5 346
  • 正文 年R本政府宣布瓤檐,位于F島的核電站,受9級特大地震影響娱节,放射性物質(zhì)發(fā)生泄漏挠蛉。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,442評論 3 331
  • 文/蒙蒙 一肄满、第九天 我趴在偏房一處隱蔽的房頂上張望谴古。 院中可真熱鬧,春花似錦稠歉、人聲如沸掰担。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,996評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽带饱。三九已至,卻和暖如春阅羹,著一層夾襖步出監(jiān)牢的瞬間勺疼,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,113評論 1 272
  • 我被黑心中介騙來泰國打工捏鱼, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留执庐,地道東北人。 一個月前我還...
    沈念sama閱讀 48,332評論 3 373
  • 正文 我出身青樓导梆,卻偏偏與公主長得像轨淌,于是被迫代替她去往敵國和親迂烁。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,044評論 2 355

推薦閱讀更多精彩內(nèi)容