Python：如何将生成器的输出保存到文本文件中？

Question

我正在使用以下生成器来计算移动平均值：

import itertools
from collections import deque
    def moving_average(iterable, n=50):
    it = iter(iterable)
    d = deque(itertools.islice(it, n-1))
    d.appendleft(0)
    s = sum(d)
    for elem in it:
        s += elem - d.popleft()
        d.append(elem)
        yield s / float(n)

我可以打印生成器的输出，但我不知道如何将该输出保存到文本文件中。

x = (1,2,2,4,1,3)
avg = moving_average(x,2)
for value in avg:
    print value

当我将打印语句改为写入文件时，输出会被打印到屏幕上，文件被创建但保持为空。

提前致谢。

- user2736211

---

参见：http://stackoverflow.com/a/6613322/1959808 - 0 _

Answer 1

def generator(howmany):
    for x in xrange(howmany):
        yield x

g = generator(10)

with open('output.txt', 'w') as f:
    for x in g:
        f.write(str(x))

with open('output.txt', 'r') as f:
    print f.readlines()

输出：

>>> 
['0123456789']

- blakev

---

谢谢！我只需要将生成器输出更改为字符串以写入文本文件。我想保持数字的列格式，所以我添加了一个行结束符：f.write(str(x)+'\n')。 - user2736211

如果您需要精确地写入生成器输出的字符串，您可以直接使用 f.writelines(g)。 - petre