用Python创建迭代流从字符串中

Question

我想用Python中的一个字符串创建一个流，这样它就相当于从文本文件中读取字符串。类似于：

for line in open('myfile.txt'): print line

除了“myfile.txt”的内容存储在字符串s中。这是正确的/最好的方法吗？

s = StringIO.StringIO("a\t\b\nc\td\n")
for line in s: print line

Answer 1

你可以用这样一个简单的generator function滚动你自己的：

def string_stream(s, separators="\n"):
    start = 0
    for end in range(len(s)):
        if s[end] in separators:
            yield s[start:end]
            start = end + 1
    if start < end:
        yield s[start:end+1]

示例用法：

>>> stream = string_stream("foo\tbar\nbaz\n", "\t\n")
>>> for s in stream:
...     print(s)
...
foo
bar
baz

cStringIO可能更快（我还没有测试过），但这将使您在定义/使用分隔符时具有灵活性。