git.net

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

generator to write N lines to file


Afternoon

Trying to create a generator to write the first N lines of text to a file.
However, I keep receiving the islice object not the text, why?

from itertools import islice

fileName = dir / "7oldsamr.txt"
dumpName = dir / "dump.txt"

def getWord(infile, lineNum):
    with open(infile, 'r+') as f:
        lines_gen = islice(f, lineNum)
        yield lines_gen

for line in getWord(fileName, 5):
    with open(dumpName, 'a') as f:
        f.write(line)

Thanks,

Sayth