StringIO学习

时间：2021-11-05 05:49:54

StringIO

StringIO的行为与file对象非常像，但它不是磁盘上文件，而是一个内存里的“文件”，我们可以将操作磁盘文件那样来操作StringIO。一个简单的例子，让你对StringIO有一个感性的认识：

from StringIO import StringIO

# 生成一个StringIO对象，当前缓冲区内容为ABCDEF

s = StringIO('ABCDEF')

# 从开头写入，将会覆盖ABC

s.write('abc')

# 每次使用read()读取前，必须seek()

# 定位到开头

s.seek(0)

# 将输出abcDEF

print s.read()

# 定位到第二个字符c

s.seek(2)

# 从当前位置一直读取到结束，将输出cDEF

print s.read()

s.seek(3)

# 从第三个位置读取两个字符，将输出DE

print s.read(2)

s.seek(6)

# 从指定位置写入

s.write('GH')

s.seek(0)

# 将输出abcDEFGH

print s.read()

# 如果读取所有内容，可以直接使用getvalue()

# 将输出abcDEFGH

print s.getvalue()

通过例子，我们看到了StringIO的行为，基本与file一致。StringIO提供了一个方法，可以方便的获取其中的数据：StringIO. getvalue()。如果使用read方法获取其中的数据，必须通过seek先设置"文件指针"的位置。

 

cStringIO

Python标准模块中还提供了一个cStringIO模块，它的行为与StringIO基本一致，但运行效率方面比StringIO更好。因此，import的时候会写成下面的方式，但使用 cStringIO模块时，有几个注意点：

1. cStringIO.StringIO不能作为基类被继承；

2. 创建cStringIO.StringIO对象时，如果初始化函数提供了初始化数据，新生成的对象是只读的。

try:

    from cStringIO import StringIO

except ImportError:

    from StringIO import StringIO

s = StringIO('A')

s.write('abc')

# 会提示：AttributeError: 'cStringIO.StringI' object has no attribute 'write'

print s.getvalue()

相关文章