python模块—StringIO and BytesIO

Posted 2020-10-13

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