将文本添加到文件的第一行

时间:2013-04-25 22:55:23

标签: python

我有一行Python代码可以产生我想要的东西。代码是:

os.system('cat {0}|egrep {1} > file.txt' .format(full,epoch))

并生成一个包含以下内容的文件:

3                321.000 53420.7046629965511    0.299                 0.00000
3                325.000 53420.7046629860714    0.270                 0.00000
3                329.000 53420.7046629846442    0.334                 0.00000
3                333.000 53420.7046629918374    0.280                 0.00000

然后我只是想调整代码,以便在顶部说“TEXT 1”,所以我尝试了第一件事,我将代码更改为:

h = open('file.txt','w')
h.write('MODE 2\n')
os.system('cat {0}|egrep {1} > file.txt' .format(full,epoch))

当我这样做时,我得到输出:

TEXT 1
   317.000 54519.6975201839344    0.627                 0.00000
3                321.000 54519.6975202038578    0.655                 0.00000
3                325.000 54519.6975201934045    0.608                 0.00000
3                329.000 54519.6975201919911    0.612                 0.00000

即。 “TEXT 1”之后的第一行不正确,并且缺少第一个“3”。任何人都可以告诉我我做错了什么,可能是一个更好的方法来完成这个简单的任务。

谢谢。

2 个答案:

答案 0 :(得分:2)

您可以按自己的方式调用grep,也可以使用subprocess.call(),这是我首选的方法。

方法1:使用os.system()

os.system('echo TEXT 1 >file.txt; egrep {1} {0} >> file.txt'.format(full, epoch))

此方法将在调用egrep之前添加 TEXT 1 。请注意,您不需要cat

方法2:使用subprocess.call()

with open('out.txt', 'wb') as output_file:
    output_file.write('TEXT 1\n')
    output_file.flush()
    subprocess.call(['egrep', epoch, full], stdout=output_file)

这是我首选的方法,原因如下:您可以更好地控制输出文件,例如在打开失败时处理异常的能力。

答案 1 :(得分:1)

你用python打开一个文件句柄,写入它,但是Python将它留给操作系统进行刷新等等 - 按照设计,如果想要在之前将某些内容写入文件其他人写了,你需要手动flush(事实上,你需要flush fsync)。

另一个注意事项:> file.txt创建一个新文件,而您可能希望追加 - 这将写为>> file.txt。简而言之:您的代码可能比您想象的更加不确定。

另一种方法,因为你已经在shell级别使用subprocess模块:

from subprocess import call

sts = call("echo 'TEXT 1' > file.txt", shell=True)
sts = call("cat {0}|egrep {1} >> file.txt".format(full,epoch), shell=True)