python - 删除标题并添加另一个

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 可以帮助您节省时间* :-)。

*(或者，花一些长的午餐时间学习更多 python 必须提供的很酷的东西!)

编码愉快!