在Python中,我們可以通過一些內置函數來操作電腦上的文件,並對文件進行讀寫,這種讀寫操作是很常見的 I/O 操作,我們今天就來簡單學習下。
open()函數
我們可以使用Python中內置的 open()
函數來打開文件,返回文件對象,並對文件進行處理。open() 函數的常見格式如下:
open(file, mode='r')
第一個參數表示要打開文件的路徑,第二個參數表示文件打開的模式,如 r 表示讀,w 表示寫。文件讀寫的常見模式如下:
文件讀寫模式 | 描述 |
---|---|
r | 用於讀,文件不存在會報錯 IOError ,如果不傳 mode 參數時默認爲該模式 |
w | 用於寫,文件不存在會自動創建 |
a | 用於追加,文件不存在會自動創建 |
b | 二進制模式 |
r+ | 相當於 r+w ,可讀可寫,如果文件不存在會報錯 IOError |
w+ | 相當於 w+r ,可讀可寫,如果文件不存在會自動創建 |
a+ | 相當於 a+r ,可追加可寫,如果文件不存在會自動創建 |
默認都是以文本模式打開文件,如果要以二進制模式打開,那麼就給對應模式加上 b
即可,如 rb、wb、ab、rb+、wb+、ab+ 等。
當然,該函數還有其他可選參數,如 encoding 、errors 等,這裏就不詳說了。
讀文件
讀文件的操作步驟爲:打開文件、讀數據、關閉文件。我們手動創建一個文本文件,隨便寫入幾行內容,然後通過代碼讀文件:
f = open("D:\\test1.txt", "r") # 以 r 模式打開文件
data = f.read() # 讀文件
f.close() # 關閉文件
在上面代碼中,調用 read()
方法可以一次讀取文件的全部內容,並存放到一個字符串中,也就是說其返回的是一個 str
類型對象。讀文件數據時,常見的函數有3個:read()、readline() 和 readlines(),它們的區別如下:
函數 | 描述 |
---|---|
read() | 每次讀取整個文件,將讀取內容放到字符串中,返回 str 對象 |
readline() | 每次只讀取文件中的一行內容,將讀取到的一行內容放到字符串中,返回 str 對象 |
readlines() | 每次按行來讀取整個文件,讀取到的每行內容依次放到列表中,返回 list 對象 |
在文件使用完畢後,因爲文件對象會佔用到操作系統的資源,所以最後必須調用 close()
來關閉文件。
寫文件
寫文件的操作步驟爲:打開文件、寫數據、關閉文件。在寫文件時,分爲覆蓋寫入和追加寫入。覆蓋寫入使用模式爲 w,相當於把文件內容刪除後重新寫入內容。
f = open("D:\\test2.txt", "w")
f.write("hello world 1\n")
f.write("hello world 2\n")
f.close()
以上代碼執行後,文件中的內容爲3行(包括最後換行):
hello world 1
hello world 2
追加寫入使用模式爲 a,它不會刪除文件內容,而是直接在原文件最後追加內容。
f = open("D:\\test2.txt", "w")
f.write("hello world 3\n")
f.write("hello world 4\n")
f.close()
f = open("D:\\test2.txt", "a")
f.write("hello world 5\n")
f.write("hello world 6\n")
f.close()
以上代碼執行後,文件中的內容爲5行(包括最後換行):
hello world 3
hello world 4
hello world 5
hello world 6
在寫文件中使用完畢後,也必須調用 close()
來關閉文件。如果我們對一個文件先執行寫操作,但忘了關閉文件,最後再對該文件執行讀操作,這樣最後讀取到的結果是什麼呢?
f1 = open("D:\\test3.txt", "w")
f1.write("hello world 1\n")
f1.write("hello world 2\n")
# f1.close()
f2 = open("D:\\test3.txt", "r")
data = f2.read()
f2.close()
print(data)
執行上面代碼後,可以發現控制檯不會打印任何內容出來,這也就說明了最後讀取不到該文件的內容。
通過 with 來讀寫文件
在文件讀寫操作時,有時候我們可能會忘了 close()
操作,或者在 close()
之前出現報錯(報錯之後的代碼就不會執行了)。所以我們需要對代碼進行優化,保證無論是否出現報錯,最後都要執行 close()
操作,我們可以通過 try ... finally
來實現:
try:
f = open("D:\\test3.txt", "w")
f.read() # w模式下沒有read(),會導致報錯
finally:
print(f.closed) # 檢查文件是否關閉,返回 False
f.close()
print(f.closed) # 檢查文件是否關閉,返回 True
雖然 try ... finally
使用起來沒什麼問題,但每次操作文件都這樣寫就顯得特別麻煩,所以,我們還可以使用 with語句
來讀寫文件。
with open("D:\\test3.txt", "w") as f1:
f1.write("hello world 1\n")
f1.write("hello world 2\n")
with open("D:\\test3.txt", "r") as f2:
data = f2.read()
print(data)
使用 with語句
的好處在於,with內的語句執行完畢後,無論是否發生異常,都會自動關閉打開的文件,這樣就不需要手動去執行 close()
操作,從而使代碼看起來更加簡潔。