刪除標題並添加另一個

Question

我必須刪除許多文件的前2個標題並將其替換爲另一個。由於我是Python和編程新手，我一直在使用下面的代碼：

import glob 
import os 
list_of_files = glob.glob('./*.txt') 
for file_name in list_of_files: 
    os.system('sed "1,2d" %s | sort -k1 > %s.sort' %(file_name,file_name)) 
    os.system ('cat header file %s.sort > %s.header' %(file_name,file_name)) 

它的工作原理。不過，我認爲應該有更好的方法來完成這件事。此外，我不必要地製作一個額外的文件* .sort，我不需要。

Answer 1

信不信由你，你可以在純Python做到這一點很容易地：

import itertools 
with open(filename) as fin: 
    ilines = itertools.islice(fin, 2, None) #this throws away the first 2 lines 
    lines = sorted(ilines, key=lambda x: x.split()[0]) #sort lexicographically on first column 

with open('header') as header, open('%s.header'%filename) as fout: 
    fout.writelines(header) #write the header 
    fout.writelines(lines) #write the data 

大功告成。稍微延長午休時間，因爲python可以幫助您節省時間* :-)。

*（或者，花一些你長的午餐學習更多的很酷的事情蟒蛇來！）

編碼快樂！

Answer 2

避免os.system。

第一種方法可能是

import glob 
import subprocess 
list_of_files = glob.glob('./*.txt') 
for file_name in list_of_files: 
    sp1 = subprocess.Popen(['sed', '1,2d', file_name], stdout=subprocess.PIPE) 
    sp2 = subprocess.Popen(['sort', '-k1'], stdin=sp1.stdout, stdout=subprocess.PIPE) 
    out = open(file_name + '.header', 'w') 
    sp3 = subprocess.Popen(['cat', 'header', 'file', '-'], stdin=sp2.stdout, stdout=out) 
    sp1.stdout.close() # sp2 got it, not our business any longer 
    sp2.stdout.close() # sp3 got it, not our business any longer 
    out.close() 
    sp1.wait() 
    sp2.wait() 
    sp3.wait()