1.StringIO模块

在平时开发过程中,有些时候我们可能不需要写在文件中,我们可以直接通过StringIO模块直接写入到系统内存中,如果不用了,可以直接清除就可以了。StringIO主要是用来在内存中写入字符串的,及字符串的缓存。


1.1通过StringIO写入内存

例子

#from io import StringIO

from io import BytesIO as StringIO

output = StringIO()

output.write("hello,world")

print(output.getvalue())

output.truncate(0)

print(output.getvalue())


结果:

hello,world


说明:

Output.write()         # 写入一个字符串

Output.getvalue()       # 用户获取写入后的字符串

Output.truncate(0)      # 参数为0,表示清空所有写入的内容


1.2通过字符串初始化一个StringIO

要读取StringIO,可以用一个str初始化StringIO,然后,像读文件一样读取

例子

#from io import StringIO

from io import BytesIO as StringIO

output = StringIO("hello\nworld\nhello\nChina")

print(output.read())

while 1:

   s = output.readline()

   if s == "":

       break

   print s.strip()


结果:

hello

world

hello

China


2.BytesIO模块

StringIO操作的只能是str,如果要操作二进制数据,就需要使用BytesIO;BytesIO实现了在内存中读写bytes,我们创建一个BytesIO,然后写入一些bytes

例子

from io import StringIO,BytesIO

f = BytesIO()

f.write(b"hello")

f.write(b"\n")

f.write("world")

print(f.getvalue())

g = BytesIO("hello\nworld")

print(g.getvalue())


结果:

hello

world

hello

world