作者:和寧世杰471 | 来源:互联网 | 2023-02-01 22:10
我想使用python删除文件中的最后几行.该文件的大小很大,所以要删除前几行我正在使用以下代码importsyswithopen(sys.argv[1],rb)asf:for_
我想使用python删除文件中的最后几行.该文件的大小很大,所以要删除前几行我正在使用以下代码
import sys
with open(sys.argv[1],"rb") as f:
for _ in range(6):#skip first 6 lines
next(f)
for line in f:
print line
解决方法:
这是一个用于截断任何可迭代的通用生成器:
from collections import deque
def truncate(iterable, num):
buffer = deque(maxlen=num)
iterator = iter(iterable)
# Initialize buffer
for n in range(num):
buffer.append(next(iterator))
for item in iterator:
yield buffer.popleft()
buffer.append(item)
truncated_range20 = truncate(range(20), 5)
print(list(truncated_range20))
# [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
使用truncate,您可以这样做:
from __future__ import print_function
import sys
from itertools import islice
filepath = sys.argv[1]
with open(filepath, 'rb') as f:
for line in truncate(islice(f, 6, None), 5):
print(line, end='')