学习Python:StringIO与cStringIO

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先设置"文件指针"的位置。

 

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()  

  

 

posted @ 2017-06-02 10:03  553490191  阅读(406)  评论(0编辑  收藏  举报