python的文件操做

文件讀寫

本文轉自廖雪峯老師的教程https://www.liaoxuefeng.com/wiki/1016959663602400/1017607179232640

讀寫文件是最多見的IO操做。Python內置了讀寫文件的函數，用法和C是兼容的。

讀文件

要以讀文件的模式打開一個文件對象，使用Python內置的open()函數，傳入文件名和標示符：

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

標示符'r'表示讀，這樣，咱們就成功地打開了一個文件。

若是文件打開成功，接下來，調用read()方法能夠一次讀取文件的所有內容，Python把內容讀到內存，用一個str對象表示：

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

最後一步是調用close()方法關閉文件。文件使用完畢後必須關閉，由於文件對象會佔用操做系統的資源，而且操做系統同一時間能打開的文件數量也是有限的：

>>> f.close()

因爲文件讀寫時都有可能產生IOError，一旦出錯，後面的f.close()就不會調用。因此，爲了保證不管是否出錯都能正確地關閉文件，咱們可使用try ... finally來實現：

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

可是每次都這麼寫實在太繁瑣，因此，Python引入了with語句來自動幫咱們調用close()方法：

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

這和前面的try ... finally是同樣的，可是代碼更佳簡潔，而且沒必要調用f.close()方法。

調用read()會一次性讀取文件的所有內容，若是文件有10G，內存就爆了，因此，要保險起見，能夠反覆調用read(size)方法，每次最多讀取size個字節的內容。另外，調用readline()能夠每次讀取一行內容，

調用readlines()一次讀取全部內容並按行返回list。所以，要根據須要決定怎麼調用。

若是文件很小，read()一次性讀取最方便；若是不能肯定文件大小，反覆調用read(size)比較保險；若是是配置文件，調用readlines()最方便：

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

二進制文件

前面講的默認都是讀取文本文件，而且是UTF-8編碼的文本文件。要讀取二進制文件，好比圖片、視頻等等，用'rb'模式打開文件便可：

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

字符編碼

要讀取非UTF-8編碼的文本文件，須要給open()函數傳入encoding參數，例如，讀取GBK編碼的文件：

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

遇到有些編碼不規範的文件，你可能會遇到UnicodeDecodeError，由於在文本文件中可能夾雜了一些非法編碼的字符。遇到這種狀況，open()函數還接收一個errors參數，表示若是遇到編碼錯誤後如何處理。最簡單的方式是直接忽略：

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

寫文件

寫文件和讀文件是同樣的，惟一區別是調用open()函數時，傳入標識符'w'或者'wb'表示寫文本文件或寫二進制文件：

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

你能夠反覆調用write()來寫入文件，可是務必要調用f.close()來關閉文件。當咱們寫文件時，操做系統每每不會馬上把數據寫入磁盤，而是放到內存緩存起來，空閒的時候再慢慢寫入。只有調用close()方法時，操做系統才保證把沒有寫入的數據所有寫入磁盤。忘記調用close()的後果是數據可能只寫了一部分到磁盤，剩下的丟失了。因此，仍是用with語句來得保險：

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

要寫入特定編碼的文本文件，請給open()函數傳入encoding參數，將字符串自動轉換成指定編碼。

若是文件已存在，會直接覆蓋（至關於刪掉後新寫入一個文件）。若是咱們但願追加到文件末尾怎麼辦？能夠傳入'a'以追加（append）模式寫入。