文件讀寫是常見的IO操作,Python讀寫文件的方法和java類似
讀文件open()
如果文件不存在,open()函數就會拋出一個IOError的錯誤,並且給出錯誤碼和詳細的信息:
f.read()
使用print()函數可以顯示f.read()內容
最後一步是調用close()方法關閉文件。文件使用完畢後必須關閉,因爲文件對象會佔用操作系統的資源,並且操作系統同一時間能打開的文件數量也是有限的:
f.close()
由於文件讀寫時都有可能產生IOError,一旦出錯,後面的f.close()就不會調用。所以,爲了保證無論是否出錯都能正確地關閉文件,我們可以使用try ... finally來實現:
for line in f.readlines():
print(line.split('_')[0]) # 按"_"分割取出第一個索引內容
像open()函數返回的這種有個read()方法的對象,在Python中統稱爲file-like Object。除了file外,還可以是內存的字節流,網絡流,自定義流等等。file-like Object不要求從特定類繼承,只要寫個read()方法就行。
要讀取非UTF-8編碼的文本文件,需要給open()函數傳入encoding參數,例如,讀取GBK編碼的文件:
寫文件
寫文件和讀文件是一樣的,唯一區別是調用open()函數時,傳入標識符'w'或者'wb'表示寫文本文件或寫二進制文件:
細心的童鞋會發現,以'w'模式寫入文件時,如果文件已存在,會直接覆蓋(相當於刪掉後新寫入一個文件)。如果我們希望追加到文件末尾怎麼辦?可以傳入'a'以追加(append)模式寫入。
例如:
讀文件open()
f = open(r'C:\Users\MARS\Desktop\remain_201805_no_uploaded.txt','r')
C:\Users\MARS\Desktop\remain_201805_no_uploaded.txt爲文件絕對路徑,r標識讀--read如果文件不存在,open()函數就會拋出一個IOError的錯誤,並且給出錯誤碼和詳細的信息:
Traceback (most recent call last):
File "C:\Users\MARS\PycharmProjects\untitled1\.idea\pythonlearning.py", line 3, in <module>
f = open(r'C:\\Users\MARS\Desktop\remain_20180_no_uploaded.txt','r')
FileNotFoundError: [Errno 2] No such file or directory: 'C:\\\\Users\\MARS\\Desktop\\remain_20180_no_uploaded.txt'
如果文件打開成功,接下來,調用read()方法可以一次讀取文件的全部內容,Python把內容讀到內存,用一個str對象表示:f.read()
使用print()函數可以顯示f.read()內容
最後一步是調用close()方法關閉文件。文件使用完畢後必須關閉,因爲文件對象會佔用操作系統的資源,並且操作系統同一時間能打開的文件數量也是有限的:
f.close()
由於文件讀寫時都有可能產生IOError,一旦出錯,後面的f.close()就不會調用。所以,爲了保證無論是否出錯都能正確地關閉文件,我們可以使用try ... finally來實現:
try:
f = open(r'C:\Users\MARS\Desktop\remain_201805_no_uploaded.txt','r')
print(f.read())
finally:
if f:
f.close()
但是每次都這麼寫實在太繁瑣,所以,Python引入了with語句來自動幫我們調用close()方法:with open(r'C:\Users\MARS\Desktop\remain_201805_no_uploaded.txt','r') as f:
print(f.read())
調用read()會一次性讀取文件的全部內容,如果文件有10G,內存就爆了,所以,要保險起見,可以反覆調用read(size)方法,每次最多讀取size個字節的內容。另外,調用readline()可以每次讀取一行內容,調用readlines()一次讀取所有內容並按行返回list。因此,要根據需要決定怎麼調用。for line in f.readlines():
print(line.split('_')[0]) # 按"_"分割取出第一個索引內容
像open()函數返回的這種有個read()方法的對象,在Python中統稱爲file-like Object。除了file外,還可以是內存的字節流,網絡流,自定義流等等。file-like Object不要求從特定類繼承,只要寫個read()方法就行。
StringIO就是在內存中創建的file-like Object,常用作臨時緩衝。
二進制文件
f = open('/Users/michael/test.jpg', 'rb')
f.read()
字符編碼要讀取非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(r'C:\Users\MARS\Desktop\haha.txt','w')
f.write('Hello, world!')
f.close()
你可以反覆調用write()來寫入文件,但是務必要調用f.close()來關閉文件。當我們寫文件時,操作系統往往不會立刻把數據寫入磁盤,而是放到內存緩存起來,空閒的時候再慢慢寫入。只有調用close()方法時,操作系統才保證把沒有寫入的數據全部寫入磁盤。忘記調用close()的後果是數據可能只寫了一部分到磁盤,剩下的丟失了。所以,還是用with語句來得保險:with open(r'C:\Users\MARS\Desktop\haha.txt','w') as f:
f.write('hello nimei')
要寫入特定編碼的文本文件,請給open()函數傳入encoding參數,將字符串自動轉換成指定編碼。細心的童鞋會發現,以'w'模式寫入文件時,如果文件已存在,會直接覆蓋(相當於刪掉後新寫入一個文件)。如果我們希望追加到文件末尾怎麼辦?可以傳入'a'以追加(append)模式寫入。
例如:
with open(r'C:\Users\MARS\Desktop\haha.txt','a') as f:
f.write('hello nimei'+'\n') #寫入時換行