StringIO的行为与file对象非常像,但它不是磁盘上文件,而是一个内存里的“文件”,我们可以像操作磁盘文件那样来操作StringIO。这篇文章主要介绍了Python StringIO模块,此模块主要用于在内存缓冲区中读写数据。模块中只有一个StringIO类,所以它的可用方法都在类中,此类中的大部分函数都与对文件的操作方法类似。
----------------------------------
s=StringIO.StringIO([buf])
此实例类似于open方法,不同的是它并不会在硬盘中生成文件,而只寄存在缓冲区;可选参数buf是一个str或unicode类型。它将会与其他后续写入的数据存放在一起(注意,若要在初始化数据之后继续写入数据,则在写入数据之前,应先将读写位置移动到结尾,然后再写入,否则,初始化数据会被覆盖掉,因为读写位置默认是0)。
StringIO类中的方法:
s.read([n])
参数n限定读取长度,int类型;缺省状态为从当前读写位置读取对象s中存储的所有数据。读取结束后,读写位置被移动。
----------------------
s.readline([length])
参数length限定读取的结束位置,int类型,缺省状态为None:从当前读写位置读取至下一个以“\n”为结束符的当前行。读写位置被移动。
----------------------
s.readlines([sizehint])
参数sizehint为int类型,缺省状态为读取所有行并作为列表返回,除此之外从当前读写位置读取至下一个以“\n”为结束符的当前行。读写位置被移动。
----------------------
s.write(s)
从读写位置将参数s写入给对象s。参数s为str或unicode类型。读写位置被移动。
----------------------
s.writelines(list)
从读写位置将list写入给对象s。参数list为一个列表,列表的成员为str或unicode类型。读写位置被移动。
----------------------
s.getvalue()
此函数没有参数,无论读写位置在哪里,都能够返回对象s中的所有数据。
----------------------
s.truncate([size])
1》有size参数
无论读写位置在哪里,都从起始位置开始,裁剪size字节的数据。
2》不带size参数
将当前读写位置之前的数据,裁剪下来。
----------------------
s.tell()
返回当前读写位置。
----------------------
s.seek(pos[,mode])
移动当前读写位置至pos处,可选参数mode为0时将读写位置移动至pos处,
为1时将读写位置从当前位置起向前或向后移动|pos|个长度,
为2时将读写位置置于末尾处再向前或向后移动|pos|个长度;
mode的默认值为0。
----------------------
s.close()
释放缓冲区,执行此函数后,数据将被释放,也不可再进行操作。
----------------------
s.isatty()
此函数总是返回0。
----------------------
s.flush()
刷新内部缓冲区。
----------------------
实例1:
[python] view plain copy
def writedata(file, msg):
file.write(msg)
f = open(r'C:\Users\91135\Desktop\test.txt', 'w')
writedata(f, "xxxxx!!!!")
f.close()
f = open(r'C:\Users\91135\Desktop\test.txt', 'r')
print f.read()
f.close()
实例2:
[python] view plain copy
import StringIO
def writedata(file, msg):
file.write(msg)
s = StringIO.StringIO('python')
print s.tell()#读写位置默认是0,因此,之后写入的数据("xxxxx!!!xxxxxx")会将之前的数据('python')覆盖掉
writedata(s, "xxxxx!!!xxxxxx")
print s.getvalue()
实例3:
[python] view plain copy
import StringIO
s = StringIO.StringIO('python')
s.seek(0,2)#将读写位置移动到结尾
s.write("aaaa")
lines = ['xxxxx', 'bbbbbbb']
s.writelines(lines)
s.write("ttttttttt")
print s.getvalue()
#如果使用read方法获取其中的数据,必须通过seek先设置"文件指针"的位置。
s.seek(0,0)#使用s.read()来读取所有数据前,应将读写位置移动到开头
print s.read()
print s.len
实例4:
[python] view plain copy
import StringIO
s = StringIO.StringIO("python")
#读写位置默认是0,下面的语句在写入数据时,并没有移动读写位置,因此,之前的数据("python")会被覆盖掉。
s.write("hello python!")
s.write('hello world!')
s.seek(0)
print s.read()
s.seek(-4,2)#移动读写位置,以便读取最后4个字节
print s.read()
通过例子,我们看到了StringIO的行为,基本与file一致。StringIO提供了一个方法,无论读写位置在哪里,都可以方便的获取其中的数据:StringIO.getvalue()。
python标准模块中还提供了一个cStringIO模块,它的行为与StringIO基本一致,但运行效率方面比StringIO更好。但使用cStringIO模块时,有几个注意点:
1. cStringIO.StringIO不能作为基类被继承;
2. 创建cStringIO.StringIO对象时,如果初始化函数提供了初始化数据,新生成的对象是只读的。所以下面的代码是错误的:
[python] view plain copy
import cStringIO
s = cStringIO.StringIO("python");
print s
print type(s)
print s.getvalue()
s.write("OOOKKK");#AttributeError: 'cStringIO.StringI' object has no attribute 'write'