提取前两排和最后一排从.txt或.csv的Python

Question

我需要从大量的.txt和.csv文件中提取的第2排和最后一排。我如何允许用户选择一个文件，并输出一个新的.txt或.csv文件，只有这3行？

Answer 1

这是你所需要的：

def extract_lines(filename,outputname): 
    l = [] 
    with open(filename,'r') as f: 
     for index,line in enumerate(f): #This iterates the file line by line which is memory efficient in case the csv is huge. 
      if index < 2: #first 2 lines 
       l.append(line) 
     if index > 1: # means the file has at least 3 lines 
      l.append(line) 
    with open(outputname,'w') as f: 
     for line in l: 
      f.write(line) 

Answer 2

我想你也可以使用bash脚本来实现这一要求。

#!/bin/bash 

for file in $(find . -name '*.txt' -o -name '*.csv') 
do 
    sed -n -e '1,2p' -e '$p' ${file} > "result"${file:(-5)} 
done 

此脚本将搜索所有以txt或csv结尾的文件。它会剪切前两行和最后一行，将这些行存储在一个新文件中。

例如，我有三个名为file1.txt，file2.txt，file3.csv的文件，它会分割每个文件的三行，并分别将它们存储在result1.txt，result2.txt，result3.csv中。

Answer 3

def get_lines(filename, front=2, rear=1): 
    result = [] 
    with open(filename, 'rb') as f: 
     for i, val in enumerate(f): 
      if i >= front: 
       break 
      result.append(val) 

     back_pos = -2 
     f.seek(back_pos, 2) # jump to the second end byte 

     rear_count = 0 
     while True: 
      if '\n' in f.read(1): 
       rear_count += 1 

      if rear_count >= rear: 
       result.extend(f.readlines()) 
       break 

      back_pos -= 1 
      f.seek(back_pos, 2) 

    return result 

很容易读第一行，但很难读最后一行。 到iter行很慢。