很多时候,数据读写不一定是文件,也可以在内存中读写。
1、StringIO
StringIO顾名思义就是在内存中读写str。
1.1、写入StringIO
要把str写入StringIO,我们需要先创建一个StringIO,然后,像文件一样写入即可。
>>> from io import StringIO # 导入StringIO类
>>> f = StringIO() # 创建一个实例,赋给f对象
>>> f.write('hello') # 往 f 中写入
5
>>> f.write(' ')
1
>>> f.write('world!')
6
>>> print(f.getvalue()) #getvalue()方法用于获得写入后的str
hello world!
1.2、读取StringIO
要读取StringIO,可以用一个str初始化StringIO,然后,像读文件一样读取
>>> from io import StringIO
>>> f = StringIO('Hello!\nHi!\nGoodbye!') #创建一个带内容的实例
>>> while True: # while循环
... s = f.readline() # 按行读取内容
... if s == '':
... break
... print(s.strip()) # strip(),删除行首行尾的空格
...
Hello!
Hi!
Goodbye!
2、BytesIO
StringIO操作的只能是str,如果要操作二进制数据,就需要使用BytesIO。
2.1、写入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'
2.2、读取BytesIO
>>> from io import BytesIO
>>> f = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87')
>>> f.read() # 只能读一次,再读为空。 可以把f.read()赋给某个变量,然后解码变量,显示值
b'\xe4\xb8\xad\xe6\x96\x87'
3、小结
StringIO和BytesIO是在内存中操作str和bytes的方法,使得和读写文件具有一致的接口。
Python学习笔记__9.2章 StringIO 和 BytesIO
原文地址:http://blog.51cto.com/12758568/2116887