Python中IO程式設計-StringIO和BytesIO
阿新 • • 發佈:2018-11-27
Python在記憶體中讀寫資料,用到的模組是StringIO和BytesIO
StringIO
>>> from io import StringIO >>> f = StringIO() >>> f.write('hello') 5 >>> f.write(' ') 1 >>> f.write('world!') 6 >>> print(f.getvalue()) hello world!
getvalue()
方法用於獲得寫入後的str。
要讀取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
StringIO操作的只能是str,如果要操作二進位制資料,就需要使用BytesIO。
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'
請注意,寫入的不是str,而是經過UTF-8編碼的bytes。
和StringIO類似,可以用一個bytes初始化BytesIO,然後,像讀檔案一樣讀取:
>>> from io import BytesIO >>> f = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87') >>> f.read() b'\xe4\xb8\xad\xe6\x96\x87'
總結
# stringIO 比如說,這時候,你需要對獲取到的資料進行操作,但是你並不想把資料寫到本地硬碟上,這時候你就可以用stringIO
>>> from io import StringIO >>> from io import BytesIO >>> def outputstring(): return 'string \nfrom \noutputstring \nfunction' >>> s = outputstring()
# 將函式返回的資料在記憶體中讀
>>> sio = StringIO(s)
# 可以用StringIO本身的方法
>>> print(sio.getvalue()) string from outputstring function
# 也可以用file-like object的方法
>>> s= sio.readlines() >>> for i in s: print(i.strip()) >>>string from outputstring function
# 將函式返回的資料在記憶體中寫
>>> sio = StringIO() >>> sio.write(s) 36
# 可以用StringIO本身的方法檢視
>>> s = sio.getvalue() >>> print(s) string from outputstring function
# 如果你用file-like object的方法檢視的時候,你會發現資料為空
>>> sio = StringIO() >>> sio.write(s) 36 >>> for i in sio.readlines(): print(i.strip()) >>>
# 這時候我們需要修改下檔案的指標位置
# 我們發現可以打印出內容了
>>> sio = StringIO() >>> sio.write(s) 36 >>> sio.seek(0,0) 0 >>> print(sio.tell()) 0 >>> for i in sio.readlines(): print(i.strip()) string from outputstring function
# 這就涉及到了兩個方法seek 和 tell
# tell 方法獲取當前檔案讀取指標的位置
# seek 方法,用於移動檔案讀寫指標到指定位置,有兩個引數,第一個offset: 偏移量,需要向前或向後的位元組數,正為向後,負為向前;第二個whence: 可選值,預設為0,表示檔案開頭,1表示相對於當前的位置,2表示檔案末尾
# 用seek方法時,需注意,如果你開啟的檔案沒有用'b'的方式開啟,則offset無法使用負值哦
# stringIO 只能操作str,如果要操作二進位制資料,就需要用到BytesIO
# 上面的sio無法用seek從當前位置向前移動,這時候,我們用'b'的方式寫入資料,就可以向前移動了
>>> bio = BytesIO() >>> bio.write(s.encode('utf-8')) 36 >>> print(bio.getvalue()) b'string \nfrom \noutputstring \nfunction' >>> bio.seek(-36,1) 0 >>> print(bio.tell()) 0 >>> for i in bio.readlines(): print(i.strip()) b'string' b'from' b'outputstring' b'function'