如何使用python在文件的第一行之前插入新行？

Question

更多详情如下：

1st line

2nd line

3rd line

4th line

...

现在要在1st line之前插入名为zero line的新行。文件如下所示：

zero line

1st line

2nd line

3rd line

4th line

...

我知道sed命令可以完成这项工作，但是如何使用python呢？谢谢

Answer 1

您可以使用fileinput

>>> import fileinput
>>> for linenum,line in enumerate( fileinput.FileInput("file",inplace=1) ):
...   if linenum==0 :
...     print "new line"
...     print line.rstrip()
...   else:
...     print line.rstrip()
...