### 一、StringIO
很多时候，数据读写不一定是文件，也可以在内存中读写。
`StringIO`顾名思义就是在内存中读写`str`。
要把`str`写入`StringIO`，我们需要先创建一个`StringIO`，然后，像文件一样写入即可：
```python
#!/usr/bin/env

from io import StringIO

# 写入内存
f = StringIO()
# write() 返回字节数
print(f.write('hello'))
print(f.write(' '))
print(f.write('world!'))

# 读取内存
print(f.getvalue())
```
`getvalue()`方法用于获得写入后的`str`。
要读取`StringIO`，可以用一个`str`初始化`StringIO`，然后，像读文件一样读取：
```python
#!/usr/bin/env

from io import StringIO

# 初始化时传入数据，\n可换行
f = StringIO('Hello!\nHi!\nGoodbye!')
while True:
    # 读取一行
    s = f.readline()
    # 没内容退出循环
    if s == '':
        break
    # 取出当前内容
    print(s.strip())
```

### 二、BytesIO
`StringIO`操作的只能是`str`，如果要操作二进制数据，就需要使用`BytesIO`。
`BytesIO`实现了在内存中读写`bytes`，我们创建一个`BytesIO`，然后写入一些`bytes`：
```pyhton
#!/usr/bin/env

from io import BytesIO

# 写入内存
f = BytesIO()
# write()，返回字节数
print(f.write('中文'.encode('utf-8')))
# 读取内存
print(f.getvalue())
```
请注意，写入的不是`str`，而是经过`UTF-8`编码的`bytes`。
和`StringIO`类似，可以用一个`bytes`初始化`BytesIO`，然后，像读文件一样读取：
```python
#!/usr/bin/env

from io import BytesIO

# 初始化时写入数据
f = BytesIO(b'\xe4\xb8\xad\xe6\x96\x87')
# 读取内存
print(f.read())
```

### 小结
`StringIO`和`BytesIO`是在内存中操作`str`和`bytes`的方法，使得和读写文件具有一致的接口。

已有0条评论

添加新评论☺

如果认为本文对您有所帮助请赞助本站

支付宝扫一扫赞助

已有0条评论

添加新评论☺