1. 程式人生 > >Python StringIO和BytesIO

Python StringIO和BytesIO

(廖雪峰教程筆記)

1、StringIO

很多時候,資料讀寫不一定是檔案,也可以在記憶體中讀寫。
StringIO顧名思義就是在記憶體中讀寫str。
要把str寫入StringIO,我們需要先建立一個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!

2、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和BytesIO是在記憶體中操作str和bytes的方法,使得和讀寫檔案具有一致的介面。

在記憶體中讀寫資料,實際中到底在哪可以用到呢?
比如你用別人寫的函式,該函式可以從檔案中讀取一個很大的矩陣。而你的應用中這個矩陣是動態生成的,你可以先把這個矩陣寫在一個StringIO裡面然後傳遞給該函式。