1. 程式人生 > >Python中IO程式設計-StringIO和BytesIO

Python中IO程式設計-StringIO和BytesIO

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'
複製程式碼