Python：将文件截断为100行或更少

Question

我刚接触Python，想知道有没有什么库函数或者简单的方法可以把一个文件截断到前100行或者更少的行数？

Answer 1

你可以参考这里的解决方案：逐行遍历字符串，然后只取前面的一百行，也就是：

import itertools
lines = itertools.islice(iter, 100)

Answer 2

with open("my.file", "r+") as f:
    [f.readline() for x in range(100)]
    f.truncate()

with open("my.file", "r+") as f:
    for x in xrange(100):
        f.readline()
    f.truncate()

编辑使用xrange迭代器而不是存储整个列表，可以提高5%的速度：

2 个回答