Python基础-StringIO和BytesIO

StringIO

它主要是用在内存读写str中。

主要用法就是：

from io import StringIO

f = StringIO()
f.write('12345')
print(f.getvalue())

f.write('54321')
f.write('abcde')

print(f.getvalue())

#打印结果
12345
1234554321abcde

也可以使用str初始化一个StringIO然后像文件一样读取。

f = StringIO('hello\nworld!')
while True:
    s = f.readline()
    if s == '':
        break
    print(s.strip()) #去除\n
#打印结果
hello
world!

BytesIO

想要操作二进制数据，就需要使用BytesIO。
当然包括视频、图片等等。

from io import BytesIO

f = BytesIO()
f.write('保存中文'.encode('utf-8'))

print(f.getvalue())
#打印结果
b'\xe4\xbf\x9d\xe5\xad\x98\xe4\xb8\xad\xe6\x96\x87'

请注意，写入的不是str，而是经过UTF-8编码的bytes。

存放图片

f = BytesIO()

image_open = open('./1.jpg', 'rb')
f.write(image_open.read())

image_save = open('./2.jpg', 'wb')
image_save.write(f.getvalue())