1. 程式人生 > >重新學習python系列(三)? WTF?

重新學習python系列(三)? WTF?

nal open python 所有 errors 測試 brush 文件 user

讀取文件:

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

-------------------------------------

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

-------------------------------------

調用readline()可以每次讀取一行內容,調用readlines()一次讀取所有內容並按行返回list

f = open(‘/path/to/file‘, ‘r‘)
for line in f.readlines():
    print(line.strip()) # 把末尾的‘\n‘刪掉

寫文件

寫文件和讀文件是一樣的,唯一區別是調用open()函數時,傳入標識符‘w‘或者‘wb‘表示寫文本文件或寫二進制文件:

f = open(‘/Users/michael/test.txt‘, ‘w‘)
f.write(‘Hello, world!‘)
f.close()
---------------------------
with open(‘/Users/michael/test.txt‘, ‘w‘) as f:
    f.write(‘Hello, world!‘)

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

file-like Object

open()函數返回的這種有個read()方法的對象,在Python中統稱為file-like Object。除了file外,還可以是內存的字節流,網絡流,自定義流等等。file-like Object不要求從特定類繼承,只要寫個read()方法就行。

StringIO就是在內存中創建的file-like Object,常用作臨時緩沖。

二進制文件

前面講的默認都是讀取文本文件,並且是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‘)

寫入內存 StringIO

getvalue()方法用於獲得寫入後的str。

from io import StringIO
f = StringIO()
f.write(u‘hello‘)
f.write(u‘ ‘)
f.write(u‘world‘)
print f.getvalue()

要讀取StringIO,可以用一個str初始化StringIO,然後,像讀文件一樣讀取:
>>> from io import StringIO >>> f = StringIO(‘Hello!\nHi!\nGoodbye!‘) >>> while True: ... s = f.readline() ... if s == ‘‘: ... break ... print(s.strip()) ... Hello! Hi! Goodbye!

BytesIO實現了在內存中讀寫bytes,我們創建一個BytesIO,然後寫入一些bytes:

>>> from io import BytesIO
>>> f = BytesIO()
>>> f.write(‘中文‘.encode(‘utf-8‘))
6
>>> print(f.getvalue())
b‘\xe4\xb8\xad\xe6\x96\x87‘

可以用一個bytes初始化BytesIO,然後,像讀文件一樣讀取

>>> from io import BytesIO
>>> f = BytesIO(b‘\xe4\xb8\xad\xe6\x96\x87‘)
>>> f.read()
b‘\xe4\xb8\xad\xe6\x96\x87‘

 

重新學習python系列(三)? WTF?