前言
養成一個好的習慣只需要堅持21天,Day12
IO進程
- 文件讀寫
讀寫文件前,我們先必須瞭解一下,在磁盤上讀寫文件的功能都是由操作系統提供的,現代操作系統不允許普通的程序直接操作磁盤,所以,讀寫文件就是請求操作系統打開一個文件對象(通常稱爲文件描述符),然後,通過操作系統提供的接口從這個文件對象中讀取數據(讀文件),或者把數據寫入這個文件對象(寫文件)。
讀文件
使用Python的內置函數open()
,傳入文件名和標識符:
f = open('/Users/Comme_moi/Desktop/test.rtf','r')
標誌符’r’表示讀操作,這樣就成功打開了一個文件。如果文件不存在,open()
函數就會拋出IOError
的錯誤,並且給出錯誤的詳細信息和不存在,比如:FileNotFoundError: [Errno 2] No such file or directory: '/Users/Comme_moi/Desktop/test'
。
如果文件打開成功,調用read()
方法可以一次讀取文件的全部內容,Python把內容讀到內存,用一個str對象表示:
f.read()
# 'Hello, world!'
最後一步調用f.close()
關閉文件。文件使用完畢後必須關閉,因爲文件對象會佔用操作系統的資源,並且操作系統同一時間能打開的文件數量也是有限的:
f.close()
如果文件讀寫時出現錯誤,後面的f.close就不會調用。所以爲了保證無論是否出錯都能正確地關閉文件,就可以使用try...finally
:
try:
f = open('/Users/Comme_moi/Desktop/test.rtf','r')
print(f.read())
finally:
if f:
f.close()
一般的,Python中我們引入with
語句來自動調用close()
方法:
with open('/Users/Comme_moi/Desktop/test.rtf',‘r’)as f:
print(f.read())
另外,調用readline()
可以每次讀取一行內容,調用readlines()
一次讀取所有內容並按行返回list:
for line in f.readlines():
print(line.strip()) # 把末尾的'\n'刪掉
二進制文件
上面都是默認讀取文本文件,並且文本文件是用UTF-8編碼的。要讀取二進制文件,比如視頻、圖片的時候,我們就要使用‘rb’模式打開文件,否則會出錯:
f = open('/Users/Comme_moi/Desktop/img.png','rb')
print(f.read())
# b'\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x00G\x00\x00\x005\x08\x06\...
字符編碼
要讀取非UTF-8編碼的文本文件,需要給open()
函數傳入encoding
參數,例如,讀取GBK編碼的文件:
>>> f = open('/Users/Comme_moi/Desktop/img.rtf', 'r', encoding='gbk')
>>> f.read()
# '測試'
如果遇到UnicodeDecodeError
,可以添加一個errors
參數,直接忽略:
f = open('/Users/Comme_moi/Desktop/img.rtf', 'r', encoding='gbk', errors='ignore')
寫文件
寫文件和讀文件相似,區別是傳入標誌符'w'
或者'wb'
表示寫文本文件或寫二進制文件:
with open('/Users/Comme_moi/Desktop/test.rtf','w') as f:
f.write('Test succed!')
要寫入特定編碼的文本文件,請給open()
函數傳入encoding
參數,將字符串自動轉換成指定編碼。