时间:2022-10-25 09:32:31 | 栏目:Python代码 | 点击:次
很多时候,数据读写不一定是文件,也可以在内存中读写。
要把str写入StringIO,我们需要先创建一个StringIO,然后,像文件一样写入即可:
getvalue()
方法用于获得写入后的str。
from io import StringIO f = StringIO() f.write('hello') f.write(' ') f.write('world!') print(f.getvalue()) #hello world!
要读取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!
StringIO操作的只能是str,如果要操作二进制数据,就需要使用BytesIO。
BytesIO实现了在内存中读写bytes,我们创建一个BytesIO,然后写入一些bytes:
请注意,写入的不是str,而是经过UTF-8编码的bytes。
from io import BytesIO f = BytesIO() f.write('中文'.encode('utf-8')) print(f.getvalue()) # b'\xe4\xb8\xad\xe6\x96\x87'
和StringIO类似,可以用一个bytes初始化BytesIO,然后,像读文件一样读取:
from io import BytesIO f = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87') f.read().decode('utf-8') # '中文'
StringIO和BytesIO是在内存中操作str和bytes的方法,使得和读写文件具有一致的接口。