在内存中读写数据

1、StringIO


from io import StringIO

## 创建_io.StringIO内存缓冲器对象f
f = StringIO()

## 往f中写入字符串数据,并返回字符串的长度
f.write('hello')  # 结果:5
f.write(' ')      # 结果:1
f.write('world!') # 结果:6

##提取对象f中的全部内容
f.getvalue()      # 结果:hello world!

## 要读取StringIO,可以用一个str初始化StringIO,然后,像读文件一样读取:
f1 = StringIO('Hello!\nHi!\nGoodbye!')
while True:
    s = f1.readline()
    if s == '':
        break
    print(s.strip())

结果:
Hello!
Hi!
Goodbye!


2、BytesIO


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

from io import BytesIO
## 创建BytesIO()对象
f = BytesIO()
## 往f中写入bytes数据
f.write('中文'.encode('utf-8')) 
print(f.getvalue())

## 和StringIO类似,可以用一个bytes初始化BytesIO,然后,像读文件一样读取:
f1 = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87')
f1.read() # 结果:b'\xe4\xb8\xad\xe6\x96\x87'

 

你可能感兴趣的:(Python学习笔记)