Python:如何将生成器输出保存到文本文件中?

时间:2013-08-31 20:56:36

标签: python printing save generator

我使用以下生成器来计算移动平均线:

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

当我更改打印行以写入文件时,输出将打印到屏幕上,创建一个文件但它仍然为空。

提前致谢。

1 个答案:

答案 0 :(得分:15)

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']