2

我必须删除许多文件的前 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 。

4

2 回答 2

3

信不信由你，你可以在纯 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 提供的很酷的东西！）

快乐编码！

于 2012-12-14T15:59:31.763 回答
1

避免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()
于 2012-12-14T15:36:02.027 回答